nn_pairwise_distance {torch} | R Documentation |
Pairwise distance
Description
Computes the batchwise pairwise distance between vectors ,
using the p-norm:
Usage
nn_pairwise_distance(p = 2, eps = 1e-06, keepdim = FALSE)
Arguments
p |
(real): the norm degree. Default: 2 |
eps |
(float, optional): Small value to avoid division by zero. Default: 1e-6 |
keepdim |
(bool, optional): Determines whether or not to keep the vector dimension. Default: FALSE |
Details
Shape
Input1:
where
D = vector dimension
Input2:
, same shape as the Input1
Output:
. If
keepdim
isTRUE
, then.
Examples
if (torch_is_installed()) {
pdist <- nn_pairwise_distance(p = 2)
input1 <- torch_randn(100, 128)
input2 <- torch_randn(100, 128)
output <- pdist(input1, input2)
}
[Package torch version 0.13.0 Index]