nnf_conv_tbc: Conv_tbc

In torch: Tensors and Neural Networks with 'GPU' Acceleration

Conv_tbc

Description

Applies a 1-dimensional sequence convolution over an input sequence. Input and output dimensions are (Time, Batch, Channels) - hence TBC.

Usage

nnf_conv_tbc(input, weight, bias, pad = 0)

Arguments

input	input tensor of shape (\mbox{sequence length} \times batch \times \mbox{in\_channels})
weight	filter of shape (\mbox{kernel width} \times \mbox{in\_channels} \times \mbox{out\_channels})
bias	bias of shape (\mbox{out\_channels})
pad	number of timesteps to pad. Default: 0

torch documentation built on June 7, 2023, 6:19 p.m.