pytroch小白,今天看到别人代码有exp_(), add_()等函数,一开始以为是python的,搜了半天没搜到。回来意识到是torch的,再搜发现搜的都是 VAE的定义。后来去中文手册搜了以下,发现这些是函数inplace的简写。举个例子,就明白了
a = torch.tensor([2, 2])
print("a =", a)
a.add_(1)
print("a =", a)
b = a.add_(1)
print("a =", a)
print("b =", b)
d = a.add(1)
print("a =", a)
print("d =", d)
其结果为:
可以发现当使用add_()的时候a的值被修改了,而使用add()时没有被修改。