yuanhou4333的博客

原创 张量 tensor

张量（tensor）是一种类似于数组和矩阵的特殊的数据结构。PyTorch 中模型的输入、输出和参数都是张量。张量类似 NumPy 的多维数组，但是张量可以借助像 GPU 这样的硬件加速运算。事实上，为了避免复制数据的开销，张量和 NumPy 数组经常共享底部内存。张量也针对自动微分进行了优化。下面开始 show the codes！