昇腾社区首页
中文
注册
开发者
下载

(beta)torch_npu.npu_bmmV2

须知:[object Object] 该接口计划废弃,可以使用torch.bmmtorch.view接口进行替换。

产品支持情况

产品 是否支持
[object Object]Atlas A3 训练系列产品[object Object]
[object Object]Atlas A2 训练系列产品[object Object]
[object Object]Atlas 训练系列产品[object Object]
[object Object]Atlas 推理系列产品[object Object]

功能说明

将矩阵“a”乘以矩阵“b”,生成“a*b”。支持FakeTensor模式。

函数原型

[object Object]

参数说明

  • selfTensor):2D或更高维度矩阵张量。数据类型支持float16float32int32。格式支持[ND,NHWC,FRACTALNZ][ND, NHWC, FRACTAL_NZ]
  • mat2Tensor):2D或更高维度矩阵张量。数据类型支持float16float32int32。格式支持[ND,NHWC,FRACTALNZ][ND, NHWC, FRACTAL_NZ]
  • output_sizesList[int]):默认值为[],输出的shape,用于matmul的反向传播。

调用示例

示例一:

[object Object]

示例二:

[object Object]