torch.sub

torch.sub 是 PyTorch 中用于执行逐元素减法的函数。它可以用于张量之间的减法运算，或者用来从一个张量中减去一个标量。

以下是一些使用 torch.sub 的示例：

示例 1：两个张量之间的减法

import torch

# 创建两个张量
tensor1 = torch.tensor([5.0, 6.0, 7.0])
tensor2 = torch.tensor([2.0, 3.0, 4.0])

# 执行减法运算
result = torch.sub(tensor1, tensor2)

print(result)  # 输出: tensor([3., 3., 3.])

示例 2：使用 out 参数指定输出张量

import torch

# 创建两个张量
tensor1 = torch.tensor([5.0, 6.0, 7.0])
tensor2 = torch.tensor([2.0, 3.0, 4.0])

# 创建一个用于存储结果的张量
output = torch.empty(3)

# 执行减法运算，并将结果存储在指定的输出张量中
torch.sub(tensor1, tensor2, out=output)

print(output)  # 输出: tensor([3., 3., 3.])