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()時沒有被修改。