torch_cross: Cross

In torch: Tensors and Neural Networks with 'GPU' Acceleration

Cross

Description

Cross

Usage

torch_cross(self, other, dim = NULL)

Arguments

self	(Tensor) the input tensor.
other	(Tensor) the second input tensor
dim	(int, optional) the dimension to take the cross-product in.

cross(input, other, dim=-1, out=NULL) -> Tensor

Returns the cross product of vectors in dimension dim of input and other.

input and other must have the same size, and the size of their dim dimension should be 3.

If dim is not given, it defaults to the first dimension found with the size 3.

Examples

if (torch_is_installed()) {

a = torch_randn(c(4, 3))
a
b = torch_randn(c(4, 3))
b
torch_cross(a, b, dim=2)
torch_cross(a, b)
}

torch documentation built on May 29, 2024, 9:54 a.m.