5 个PyTorch 中的处理张量的基本函数

能够以准确有效的方式构建神经网络是招聘人员在深度学习工程师中最受追捧的技能之一。PyTorch 是一个 主要用于深度学习的Python 库。PyTorch 最基本也是最重要的部分之一是创建张量，张量是数字、向量、矩阵或任何 n 维数组。在构建神经网络时为了降低计算速度必须避免使用显式循环，我们可以使用矢量化操作来避免这种循环。在构建神经网络时，足够快地计算矩阵运算的能力至关重要。

“为什么不使用 NumPy 库呢？”

对于深度学习，我们需要计算模型参数的导数。PyTorch 提供了在反向传播时跟踪导数的能力而 NumPy 则没有，这在Pytorch中被称为“Auto Grad”。PyTorch 为使用 GPU 的快速执行提供了内置支持。这在训练模型方面至关重要。由于 Numpy 缺乏将其计算转移到 GPU 的能力，因此训练模型的时间最终会变得非常大。

所有使用 PyTorch 的深度学习项目都从创建张量开始。让我们看看一些必须知道的函数，它们是任何涉及构建神经网络的深度学习项目的支柱。

• torch.tensor()
• torch.sum()
• torch.index_select()
• torch.stack()
• torch.mm()

在安装完Pytorch后，在代码中可以直接导入：

 

1. # Import torch and other required modules
2. import torch

 

完整文章

 

5 个PyTorch 中的处理张量的基本函数