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.])