pytorch和numpy默認數據類型的區別

  • numpy默認數據類型float64 (int32 for 整數)
  • PyTorch默認數據類型float32,通常需要將numpy數據的數據類型轉爲np.float32

參考:
https://blog.csdn.net/yyb19951015/article/details/84781434

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