torch.nn的loss实例 和 torch.randn(100, 100)的随机种子设置

仅作为记录，大佬请跳过。

直接上程序

import torch
from torch.autograd import Variable
loss_f = torch.nn.MSELoss()  #不用传入任何参数

torch.manual_seed(0)
# torch.cuda.manual_seed(0)
'''随机生成两个维度都是（100,100）的参数'''
x = Variable(torch.randn(100, 100))
y = Variable(torch.randn(100, 100))

loss = loss_f(x, y)  #在使用实例时需要输入两个维度一样的参数方可进行计算

loss_f2 = torch.nn.L1Loss()
loss2 = loss_f2(x, y)

print(loss)
print(loss2)

展示

参考

传送门

---

torch.randn(100, 100)的随机种子设置

参考

传送门