tensor 中mul_,add_解读

pytorch 中文网文档链接 https://ptorch.com/docs/1/Tensor

每一个张量tensor都有一个相应的torch.Storage保存其数据,张量类提供了一个多维的,横向视图的存储,并定义了数字操作。

img = ToTensor(img)
img = img.mul_(2).add_(-1)

torch.FloatTensor.abs_()会在原地计算绝对值并返回修改的张量,

tensor.FloatTensor.abs()将会在新张量中计算结果。

mul_(value)

mul()的直接运算形式,即直接执行并且返回修改后的张量

add_(value)

add()的直接运算形式,即直接执行并且返回修改后的张量

 

原文地址:https://www.cnblogs.com/shuangcao/p/11933414.html