torch.cross
torch.cross(input, other, dim=None, *, out=None) -> Tensor
Returns the cross product of vectors in dimension "dim" of "input"
   and "other".
Supports input of float, double, cfloat and cdouble dtypes. Also
   supports batches of vectors, for which it computes the product
   along the dimension "dim". In this case, the output has the same
   batch dimensions as the inputs.
If "dim" is not given, it defaults to the first dimension found
   with the size 3. Note that this might be unexpected.
See also:
 "torch.linalg.cross()" which requires specifying dim (defaulting
 to -1).

Warning:
 This function may change in a future PyTorch release to match the
 default behaviour in "torch.linalg.cross()". We recommend using
 "torch.linalg.cross()".

Parameters:
      * input (Tensor) -- the input tensor.
  * **other** (*Tensor*) -- the second input tensor

  * **dim** (*int**, **optional*) -- the dimension to take the
