pytorch exp_(), add_()等運算

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

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章