(beta)torch_npu.npu_bmmV2
须知:[object Object] 该接口计划废弃,可以使用
torch.bmm和torch.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]
参数说明
- self(
Tensor):2D或更高维度矩阵张量。数据类型支持float16、float32、int32。格式支持。 - mat2(
Tensor):2D或更高维度矩阵张量。数据类型支持float16、float32、int32。格式支持。 - output_sizes(
List[int]):默认值为[],输出的shape,用于matmul的反向传播。
调用示例
示例一:
[object Object]
示例二:
[object Object]