torch中相减的运算

在PyTorch中，可以使用torch.sub()函数对两个张量进行相减运算。下面是一个示例：

import torch

# 创建两个张量
a = torch.tensor([1, 2, 3])
b = torch.tensor([4, 5, 6])

# 相减运算
result = torch.sub(a, b)

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

在上述示例中，我们使用torch.sub()函数对张量a和b进行相减运算，得到结果为[-3, -3, -3]的张量。