Pytorch中的四则运算高级写法(机器学习)

今天发现了一种在Pytorch中可以在计算后自动修改元素值的写法，这里以四则运算为例来说明这个问题。

import torch

# 带有下划线的方法，指代就地修改，计算完成后对操作的tensor变量进行修改
t1 = torch.tensor(5)
t2 = torch.tensor(5)
t3 = torch.tensor(1)
t1.add_(t3)
print("完成加和运算后t1的值为： ")
print(t1)
print("完成乘法运算后t2的值为： ")
t2.mul_(t3)
print(t2)
"""
输出结果：
完成加和运算后t1的值为： 
tensor(6)
完成乘法运算后t2的值为： 
tensor(5)
"""

我们可以看到当计算完成后t1,t2的值均发生了变化。因为减法与加法，乘法与除法互为逆运算，因此Pytorch只提供了tensor.add_()方法和tensor.mul_()方法。

码字不易，如果大家觉得有用，请高抬贵手给一个赞让我上推荐让更多的人看到吧~