PyTorch常用的包有哪些,作用是什麼?

PyTorch常用的包有哪些,作用是什麼? 已經九月份了,暑假課程《深度學習》的學習也接近尾聲了,在這個課程中,我們做了很多基於pytorch框架的實驗,也瞭解到許多pytorch的包,下面對常用的包做一個歸納總結。 PyTorch本身是一個基於Python的科學計算庫,特點是可以在GPU上運算。 1、torch: 張量的有關運算。如創建、索引、連接、轉置、加減乘除、切片等。 2、torch.nn
相關文章
相關標籤/搜索