PyTorch------nn.ReLU(inplace = True)詳解

  • nn.ReLU我們常常在使用,但是這個inplace的參數使用的就不多了,有什麼含義和作用呢?
  • nn.ReLU()函數默認inplace 默認是False
  • 在這裏插入圖片描述
  • 測試方法如下:
  • 在這裏插入圖片描述
    inplace = False
  • 注意輸出對象地址
  • inplace = False 時,不會修改輸入對象的值,而是返回一個新創建的對象,所以打印出對象存儲地址不同,類似於C語言的值傳遞
    在這裏插入圖片描述
  • inplace = True
  • 注意輸出對象地址
  • inplace = True 時,會修改輸入對象的值,所以打印出對象存儲地址相同,類似於C語言的址傳遞,
  • 在這裏插入圖片描述
  • inplace = True ,會改變輸入數據的值,節省反覆申請與釋放內存的空間與時間,只是將原來的地址傳遞,效率更好
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章