（beta）torch_npu.npu_linear

产品支持情况

产品	是否支持
Atlas A3 训练系列产品	√
Atlas A2 训练系列产品	√
Atlas 推理系列产品	√
Atlas 训练系列产品	√

功能说明

将矩阵“a”乘以矩阵“b”，生成“a*b”。

函数原型

torch_npu.npu_linear(input, weight, bias=None) -> Tensor

参数说明

• input (Tensor)：必选参数，2D矩阵张量。数据类型支持float32、float16、int32、int8。格式支持[ND, NHWC, FRACTAL_NZ]。
• weight (Tensor)：必选参数，2D矩阵张量。数据类型支持float32、float16、int32、int8。格式支持[ND, NHWC, FRACTAL_NZ]。
• bias (Tensor)：可选参数，1D张量。数据类型支持float32、float16、int32。格式支持[ND, NHWC]。默认值为None。

调用示例

>>> import torch
>>> import torch_npu
>>> x=torch.rand(2,16).npu()
>>> w=torch.rand(4,16).npu()
>>> b=torch.rand(4).npu()
>>> output = torch_npu.npu_linear(x, w, b)
>>> output
tensor([[3.6335, 4.3713, 2.4440, 2.0081],
        [5.3273, 6.3089, 3.9601, 3.2410]], device='npu:0')