torch基本功能介紹

torch的基本功能 torch:張量的相關運算,eg:創建、索引、切片、連續、轉置、加減乘除等相關運算。 torch.nn:包含搭建網絡層的模塊(modules)和一系列的loss函數。eg.全連接、卷積、池化、BN分批處理、dropout、CrossEntropyLoss、MSLoss等。 torch.autograd:提供Tensor所有操作的自動求導方法。 torch.nn.functio
相關文章
相關標籤/搜索