mindspore.ops.Cross
- class mindspore.ops.Cross(dim=- 65530)[源代码]
返回 x1 和 x2 沿着维度 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 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]