Pytorch张量计算详解

张量基础运算

张量基础的四则运算加减乘除运算分别对应add(),sub(),mul(),div(),这四种运算函数具有广播机制

import torch

data1 = torch.tensor([1, 2, 3])
print(data1.add(2))
print(data1)

print(data1.sub(1))
print(data1)

print(data1.mul(2))
print(data1)

print(data1.div(2))
print(data1)
# tensor([3, 4, 5])
# tensor([1, 2, 3])
# tensor([0, 1, 2])
# tensor([1, 2, 3])
# tensor([2, 4, 6])
# tensor([1, 2, 3])
# tensor([0.5000, 1.0000, 1.5000])
# tensor([1, 2, 3])

add_(),sub_(),mul_(),div_(),在上面的四种函数上加上_后的区别为会在四则运算后给自身赋值，也就是+=,-=,*=,/=。

import torch

data1 = torch.tensor([1, 2, 3])
data1.add_(2)
print(data1)

data1.sub_(1)
print(data1)

data1.mul_(2)
print(data1)

data1.div_(2)
print(data1)
# tensor([3, 4, 5])
# tensor([2, 3, 4])
# tensor([4, 6, 8])
# Traceback (most recent call last):
#   File "D:\Pythonproject\teach_day_01\demo01.py", line 13, in <module>
#     data1.div_(2)