开发者
下载
[object Object][object Object][object Object]
  • Ascend 950PR/Ascend 950DT:支持
[object Object][object Object]
  • Atlas A3 训练系列产品/Atlas A3 推理系列产品:支持
[object Object][object Object]
  • Atlas A2 训练系列产品/Atlas A2 推理系列产品:支持
[object Object][object Object]

该接口为定制接口,约束较多。不保证稳定性。

该算子执行转置批量矩阵乘法。具体操作为:

  1. 将输入张量[object Object]从形状(M, B, K)转置为(B, M, K)。
  2. 执行批量矩阵乘法,将转置后的[object Object] (B, M, K)与[object Object] (B, K, N)相乘,得到中间结果(B, M, N)。
  3. 将中间结果转置回形状(M, B, N)作为最终输出。
[object Object]
[object Object]
[object Object][object Object]undefined
[object Object]

返回输出Tensor,Tensor的数据类型由[object Object]指定,形状为(M, B, N)。

[object Object]
[object Object]