pytorch(9)-item()

使用積累 item() item() 一個Tensor調用.item()方法就可以返回這個Tensor 對應的標準python 類型的數據.注意事項,只針對一個元素的標量.若是向量,可以調用tolist()方法.
本站公眾號
   歡迎關注本站公眾號,獲取更多信息