pytorch學習 基本組成、基本練習

1. 基本結構 pytorch主要分爲如下幾個模塊來訓練模型:python tensor:tensor爲基本結構,能夠直接建立,從list建立以及由numpy數組獲得,torch還提供一套運算以及shape變換方式。 Variable:自動求導機制,利用Variable包裝tensor後,即可以使用其求導的功能了,有點像個裝飾器。 nn:nn模塊是整個pytorch的核心,本身設計的Net(),繼
相關文章
相關標籤/搜索