如何理解pytorch中的tensor

it2026-01-20  9

Tensor 张量

在PyTorch中,torch.Tensor是存储和变换数据的主要工具。我们发现Tensor和NumPy的多维数组非常类似。然而,Tensor提供GPU计算和自动求梯度等更多功能,这些性能使得Tensor更加适合深度学习。

"tensor"这个单词一般可译作“张量”,张量可以看作是一个多维数组。标量可以看作是0维张量,向量可以看作1维张量,矩阵可以看作是二维张量。

本文参考了以下内容 [1] 动手学深度学习 pytorch版

最新回复(0)