nnf_conv_tbc: Conv_tbc

View source: R/nnf-conv.R

nnf_conv_tbcR Documentation

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 May 29, 2024, 9:54 a.m.