torch_cdist

Provides functionality to define and train neural networks similar to
'PyTorch' by Paszke et al (2019) <arXiv:1912.01703> but written entirely in R
using the 'libtorch' library. Also supports low-level tensor operations and
'GPU' acceleration.

Daniel Falbel

torch

Tensors and Neural Networks with 'GPU' Acceleration

Javier Luraschi

Dmitriy Selivanov

Athos Damiani

Christophe Regouby

Krzysztof Joachimiak

Hamada S. Badr

 RStudio

torch_cdist function

<dl><dt>x1</dt>
<dd>(Tensor) input tensor of shape \(B \times P \times M\).</dd>
<dt>x2</dt>
<dd>(Tensor) input tensor of shape \(B \times R \times M\).</dd>
<dt>p</dt>
<dd>NA p value for the p-norm distance to calculate between each vector pair \(\in [0, \infty]\).</dd>
<dt>compute_mode</dt>
<dd>NA 'use_mm_for_euclid_dist_if_necessary' - will use matrix multiplication approach to calculate euclidean distance (p = 2) if P &gt; 25 or R &gt; 25 'use_mm_for_euclid_dist' - will always use matrix multiplication approach to calculate euclidean distance (p = 2) 'donot_use_mm_for_euclid_dist' - will never use matrix multiplication approach to calculate euclidean distance (p = 2) Default: use_mm_for_euclid_dist_if_necessary.</dd></dl>

Arguments

Computes batched the p-norm distance between each pair of the two collections of row vectors.

TEST 

Cdist — torch_cdist

<dl>

<dt>x1</dt>
<dd>(Tensor) input tensor of shape \(B \times P \times M\).</dd>


<dt>x2</dt>
<dd>(Tensor) input tensor of shape \(B \times R \times M\).</dd>


<dt>p</dt>
<dd>NA p value for the p-norm distance to calculate between each vector pair \(\in [0, \infty]\).</dd>


<dt>compute_mode</dt>
<dd>NA 'use_mm_for_euclid_dist_if_necessary' - will use matrix multiplication approach to calculate euclidean distance (p = 2) if P &gt; 25 or R &gt; 25 'use_mm_for_euclid_dist' - will always use matrix multiplication approach to calculate euclidean distance (p = 2) 'donot_use_mm_for_euclid_dist' - will never use matrix multiplication approach to calculate euclidean distance (p = 2) Default: use_mm_for_euclid_dist_if_necessary.</dd>

</dl>

Computes batched the p-norm distance between each pair of the two collections of row vectors.

torch_cdist: Cdist

Description

Usage

Arguments

TEST