TensorPlay
API symbolsnn
Copy
View MarkdownDownload .md

tensorplay.nn.functional.channel_shuffle

tensorplay.nn.functional.channel_shuffle(input: TensorBase, groups: int) → TensorBase[source]

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.binary_cross_entropy

Compute Binary Cross Entropy between the target and input probabilities.

tensorplay.nn.functional.conv_tbc

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.

Ask DeepWiki