TensorPlay
API symbolsnn
Copy
View MarkdownDownload .md

tensorplay.nn.functional.conv_tbc

tensorplay.nn.functional.conv_tbc(input, weight, bias=None, pad=0)[source]

Applies a 1D convolution over an input of shape (T, B, C) along the time dimension (torch.conv_tbc). Expressed as a grouped (depthwise) conv1d: every channel is convolved with its own length-k filter.

tensorplay.nn.functional.channel_shuffle

Divide the channels in a tensor into g groups and rearrange them as in ShuffleNet: (*, C, H, W) -> (*, C, H, W) with channels interleaved across groups.

tensorplay.nn.functional.conv_transpose1d

Next Page

Ask DeepWiki