2024年6月24日
torch.sub
摘要： torch.sub 是 PyTorch 中用于执行逐元素减法的函数。它可以用于张量之间的减法运算，或者用来从一个张量中减去一个标量。 以下是一些使用 torch.sub 的示例： 示例 1：两个张量之间的减法 import torch # 创建两个张量 tensor1 = torch.tensor( 阅读全文
posted @ 2024-06-24 11:36 遇见生活 阅读(26) 评论(1) 推荐(1) 编辑