TensorPlay
API symbolsnn
Copy
View MarkdownDownload .md

tensorplay.nn.functional.pad

tensorplay.nn.functional.pad(input, pad, mode='constant', value=0)[source]

Pads tensor. pad values are described starting from the last dimension and moving forward, exactly like torch.nn.functional.pad. Non-constant modes support the last 3 dimensions of a 3D/4D/5D input (torch’s restriction).

tensorplay.nn.functional.one_hot

Returns long tensor shaped tensor.shape + (num_classes,) with a 1 at each label position — port of ATen one_hot.

tensorplay.nn.functional.pairwise_distance

Computes the pairwise distance between input vectors.

Ask DeepWiki