Cross
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_tensor
#> -0.0327 0.1124 -0.1158
#> -0.0779 0.3301 -0.3027
#> -1.3833 -0.0304 -0.8002
#> -0.6250 0.9997 0.1767
#> [ CPUFloatType{4,3} ]