pytorch用法記錄2

1.torch.Storage類    使用storage()函數把Tensor數據轉換爲float類型的Storage數據,再使用tolist() 返回一個包含此存儲中元素的列表。 2.detach 計算圖截斷 detach 的意思是,這個數據和生成它的計算圖「脫鉤」了,即detach就是截斷反向傳播的梯度流。GAN中,Train D on fake,G生成的數據會傳入D,然後計算loss,再反
相關文章
相關標籤/搜索