- Ascend 950PR/Ascend 950DT:支持
- Atlas A3 训练系列产品/Atlas A3 推理系列产品:支持
- Atlas A2 训练系列产品/Atlas A2 推理系列产品:支持
该接口为定制接口,约束较多。不保证稳定性。
该算子执行转置批量矩阵乘法。具体操作为:
- 将输入张量
[object Object]从形状(M, B, K)转置为(B, M, K)。 - 执行批量矩阵乘法,将转置后的
[object Object](B, M, K)与[object Object](B, K, N)相乘,得到中间结果(B, M, N)。 - 将中间结果转置回形状(M, B, N)作为最终输出。
[object Object]
返回输出Tensor,Tensor的数据类型由[object Object]指定,形状为(M, B, N)。
[object Object]