pytorch：基本数学运算

第一种：加减乘除（add、sub、mul、div）

我们以加法为例

import torch
a=torch.rand(3,4)
b=torch.rand(4)
a+b
torch.add(a,b)

可以得到对应输出结果，其中a+b得到的结果为

tensor([[1.2581, 1.3885, 0.5426, 0.3559],
        [1.4212, 0.9696, 0.3571, 1.0510],
        [0.6906, 1.1928, 0.8779, 0.4808]])

而使用torch.add（）函数进行加法，得到的结果如下：

tensor([[1.2581, 1.3885, 0.5426, 0.3559],
        [1.4212, 0.9696, 0.3571, 1.0510],
        [0.6906, 1.1928, 0.8779, 0.4808]])

为了验证是否真的一致，我们使用eq（）函数进行验证

torch.all(torch.eq(a+b,torch.add(a,b)))

返回结果为：

tensor(True)

可以看出，使用直接相加“+”和函数torch.add得到的结果一致。

对于其他运算，与加法运算使用方法一致。

第二种：matmul（矩阵相乘）

表达形式：（1）torch.mm（只适用于2d的矩阵，一般不推荐使用。）

ÿ