Cross
产品支持情况
| 产品 |
是否支持 |
| Atlas A2 训练系列产品/Atlas A2 推理系列产品 |
√ |
功能说明
out=self×other=∣ijkx1y1z1x2y2z2∣=(y1z2−y2z1)i−(x1z2−x2z1)j+(x1y2−x2y1)kout = self\times other = \begin{vmatrix}i&j&k\\x_1&y_1&z_1\\x_2&y_2&z_2\end{vmatrix} = (y_1z_2-y_2z_1)i-(x_1z_2-x_2z_1)j+(x_1y_2-x_2y_1)k
out=self×other=∣∣∣∣∣∣∣ix1x2jy1y2kz1z2∣∣∣∣∣∣∣=(y1z2−y2z1)i−(x1z2−x2z1)j+(x1y2−x2y1)k
参数说明
| 参数名 |
输入/输出/属性 |
描述 |
数据类型 |
数据格式 |
| self |
输入 |
计算公式中的输入 self。 |
FLOAT16、FLOAT、INT32、INT8、UINT8、INT16、DOUBLE、INT64、UINT16、UINT32、UINT64、COMPLEX64、COMPLEX128 |
ND |
| other |
输入 |
计算公式中的输入 other。 |
FLOAT16、FLOAT、INT32、INT8、UINT8、INT16、DOUBLE、INT64、UINT16、UINT32、UINT64、COMPLEX64、COMPLEX128 |
ND |
| out |
输出 |
计算公式中的输出 out。 |
FLOAT16、FLOAT、INT32、INT8、UINT8、INT16、DOUBLE、INT64、UINT16、UINT32、UINT64、COMPLEX64、COMPLEX128 |
ND |
- Atlas A2 训练系列产品/Atlas A2 推理系列产品:数据类型支持INT8、INT16、INT32、INT64、UINT8、FLOAT16、BFLOAT16、FLOAT、FLOAT64、COMPLEX64、COMPLEX128。
约束说明
- 指定维度(进行cross运算的维度)的size必须为3。
- 输入shape需满足broadcast关系,即两个输入tensor的shape需要兼容,能够进行广播运算。
- 输入数据类型需要保持一致。
调用说明
贡献说明
| 贡献者 |
贡献方 |
贡献算子 |
贡献时间 |
贡献内容 |
| hth810 |
个人开发者 |
Cross |
2026/4/9 |
Cross算子适配开源仓 |