（beta）torch_npu.npu_bmmV2

函数原型

功能说明

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

参数说明

• self (Tensor) - 2D或更高维度矩阵张量。数据类型：float16、float32、int32。格式：[ND, NHWC, FRACTAL_NZ]。
• mat2 (Tensor) - 2D或更高维度矩阵张量。数据类型：float16、float32、int32。格式：[ND, NHWC, FRACTAL_NZ]。
• output_sizes (ListInt，默认值为[]) - 输出的shape，用于matmul的反向传播。

支持的型号

• [object Object]Atlas 训练系列产品[object Object]
• [object Object]Atlas A2 训练系列产品[object Object]
• [object Object]Atlas A3 训练系列产品[object Object]
• [object Object]Atlas 推理系列产品[object Object]

调用示例

示例一：

示例二：