PyTorch張量
-
多維數組:張量可以是標量(0D)、向量(1D)、矩陣(2D)或更高維的數據(3D+)。
-
數據類型:支持多種數據類型(如?
float32
,?int64
,?bool
?等),通過?dtype
?指定。 -
設備支持:可以存儲在 CPU 或 GPU 上(通過?
device
?指定),例如?device='cuda'
。 -
自動微分:通過?
requires_grad=True
?標記張量,PyTorch 會跟蹤其操作以計算梯度。
PyTorch深度學習實戰(2)——PyTorch基礎_pytorch項目實戰-CSDN博客