mindspore.ops.Cross

class mindspore.ops.Cross(dim=- 65530)[源代码]

返回 x1x2 沿着维度 dim 上的向量积(叉积)。

警告

这是一个实验性API,后续可能修改或删除。

更多参考详见 mindspore.ops.cross()

参数:
  • dim (int) - 沿着此维进行叉积操作。默认值:-65530。

输入:
  • x1 (Tensor) - 输入Tensor。

  • x2 (Tensor) - 另一个输入Tensor,数据类型和shape必须和 x1 一致,并且他们的 dim 维度的长度应该为3。

输出:

Tensor,数据类型与输入相同。

支持平台:

Ascend CPU

样例:

>>> import mindspore
>>> import numpy as np
>>> from mindspore import Tensor
>>> from mindspore.common import dtype as mstype
>>> import mindspore.ops as ops
>>> cross = ops.Cross(dim = 0)
>>> x1 = Tensor([1, 2, 3], mstype.int8)
>>> x2 = Tensor([1, 2, 3], mstype.int8)
>>> output = cross(x1, x2)
>>> print(output)
[0 0 0]