torch.sub

torch.sub 是 PyTorch 中用于执行逐元素减法的函数。它可以用于张量之间的减法运算，或者用来从一个张量中减去一个标量。 以下是一些使用 torch.sub 的示例： 示例 1：两个张量之间的减法 import torch # 创建两个张量 tensor1 = torch.tensor(
posted @ 2024-06-24 11:19  Lefulann  阅读(1)  评论(0)  编辑  收藏  举报
← → ↓ ↑