pytorch多个张量的加减乘除运算

import torch

a1 = torch.rand(2, 3)
print(a1)
a2 = torch.rand(2, 3)
print(a2)

print(a1.add(a2))  # 加法运算a1 + a2，不改变参与运算的张量的值
print(a1 + a2)  # 加法运算a1 + a2，不改变参与运算的张量的值

print(a1.sub(a2))  # 减法运算a1 - a2，不改变参与运算的张量的值
print(a1 - a2)  # 减法运算a1 - a2，不改变参与运算的张量的值

print(a1.mul(a2))  # 乘法运算a1 * a2，不改变参与运算的张量的值
print(a1 * a2)  # 乘法运算a1 * a2，不改变参与运算的张量的值

print(a1.div(a2))  # 除法运算a1 / a2，不改变参与运算的张量的值
print(a1 / a2)  # 除法运算a1 / a2，不改变参与运算的张量的值

print(a1.add_(a2))  # 加法运算a1 + a2，改变a1张量的值为相加之后的张量。减、乘、除运算也一样道理
print(a1)

Done！！！