@反向傳播 / 博客 RSS 訂閱

Nov 21 2025

jowvid - Pytorch的to(device)用法-

PyTorch中，.to(device)是一個非常重要的方法，用於將張量、模型等對象移動到指定的設備（如CPU或GPU）。 import torch import torch.nn as nn # 檢查可用設備 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') print(

數據 , 顯式 , 後端開發 , 反向傳播 , harmonyos

收藏 評論