Skip to content

Blog

[PyTorch] 使用 view() 和 permute() 轉換維度

PyTorch 是一個基於 Python 的深度學習框架,我們可以藉由 PyTorch 所包裝好的模組、函式,輕易地實作我們想要實現的模型架構。而說到深度學習,就不得不提到使用 GPU 的平行化運算,提到 GPU 的平行化運算,就一定得說到我們得將輸入神經元的『維度』固定,好實現平行化運算。

Read More »[PyTorch] 使用 view() 和 permute() 轉換維度