如何在 PyTorch 中对张量执行逐元素加法？

PyTorch 是一种基于 Python 的科学计算库，用于对张量运算加速。本文将介绍如何在 PyTorch 中对张量执行逐元素加法。

什么是逐元素加法？

逐元素加法是一种将两个张量中的每个元素相加的操作。两个张量必须形状相同才能执行逐元素加法。例如，如果两个张量 $a$ 和 $b$ 都是形状为 $(n, m)$ 的矩阵，则逐元素加法 $c = a + b$ 将返回一个形状相同的矩阵 $c$，其中 $c_{i,j} = a_{i,j} + b_{i,j}$。

如何执行逐元素加法？

在 PyTorch 中，执行逐元素加法非常容易。只需使用加号运算符即可。以下是一个示例代码片段：

import torch

# 创建两个随机张量
a = torch.rand((3, 3))
b = torch.rand((3, 3))

# 执行逐元素加法
c = a + b

# 输出结果
print(c)

此代码将创建两个随机张量 $a$ 和 $b$，然后执行逐元素加法并将结果保存在 $c$ 中。最后，代码将打印结果。结果将是一个维度为 $(3, 3)$ 的随机矩阵，其元素是 $a$ 和 $b$ 的对应元素之和。

在以上代码中，我们使用 PyTorch 提供的 torch.rand 函数来创建两个随机张量。然后，我们执行逐元素加法并将结果保存到一个叫做 $c$ 的新张量中。最后，我们使用 print 函数将 $c$ 中的元素打印出来。

小结

在 PyTorch 中，执行逐元素加法是一项容易任务。只需要使用加号运算符就可以将两个张量中的每个元素相加。如果两个张量不具有相同的形状，则无法执行此操作。