TensorPlay
API symbolsnn
Copy
View MarkdownDownload .md

tensorplay.nn.functional.pairwise_distance

tensorplay.nn.functional.pairwise_distance(x1: TensorBase, x2: TensorBase, p: float = 2.0, eps: float = 1e-06, keepdim: bool = False) → TensorBase[source]

Computes the pairwise distance between input vectors.

Port of aten::pairwise_distance: norm(x1 - x2 + eps, p) over the last dimension.

tensorplay.nn.functional.pad

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

Computes the pairwise distance between rows of input .

Ask DeepWiki