torch_conv1d function

Conv1d

Conv1d

torch_conv1d( input, weight, bias = list(), stride = 1L, padding = 0L, dilation = 1L, groups = 1L )

Arguments

  • input: input tensor of shape (\mboxminibatch,\mboxin_channels,iW)(\mbox{minibatch} , \mbox{in\_channels} , iW)
  • weight: filters of shape (\mboxout_channels,\mboxin_channels\mboxgroups,kW)(\mbox{out\_channels} , \frac{\mbox{in\_channels}}{\mbox{groups}} , kW)
  • bias: optional bias of shape (\mboxout_channels)(\mbox{out\_channels}). Default: NULL
  • stride: the stride of the convolving kernel. Can be a single number or a one-element tuple (sW,). Default: 1
  • padding: implicit paddings on both sides of the input. Can be a single number or a one-element tuple (padW,). Default: 0
  • dilation: the spacing between kernel elements. Can be a single number or a one-element tuple (dW,). Default: 1
  • groups: split input into groups, \mboxin_channels\mbox{in\_channels} should be divisible by the number of groups. Default: 1

conv1d(input, weight, bias=NULL, stride=1, padding=0, dilation=1, groups=1) -> Tensor

Applies a 1D convolution over an input signal composed of several input planes.

See nn_conv1d() for details and output shape.

Examples

if (torch_is_installed()) { filters = torch_randn(c(33, 16, 3)) inputs = torch_randn(c(20, 16, 50)) nnf_conv1d(inputs, filters) }
  • Maintainer: Daniel Falbel
  • License: MIT + file LICENSE
  • Last published: 2025-02-14