代码
import torch
print('# 1、算法操作')
# 例1、加法
x = torch.rand(3, 3)
y = torch.rand(3, 3)
print(x)
print(y)
result = x + y # 加法,法1
print(result, '\n')
# result = torch.add(x, y) # 加法,法2
# print(result)
# y.add_(x) # 法3,将x加到y上
# print(y)
# 其他运算类似
# 加法 torch.add
# 减法 torch.sub
# 乘法 torch.mul
# 除法 torch.div
# 求余 torch.remainder
运行结果
# 1、算法操作
tensor([[0.7251, 0.6412, 0.8704],
[0.2415, 0.8661, 0.4362],
[0.1130, 0.8149, 0.4534]])
tensor([[0.4931, 0.0180, 0.9155],
[0.5586, 0.2503, 0.9086],
[0.1073, 0.0146, 0.3479]])
tensor([[1.2183, 0.6592, 1.7859],
[0.8001, 1.1164, 1.3447],
[0.2204, 0.8295, 0.8013]])
您可以选择一种方式赞助本站
支付宝扫一扫赞助
微信钱包扫描赞助
赏