torch.nn.Linear和torch.nn.MSELoss

分类: 文章 • 2024-01-01 16:05:22

1.torch.nn.Linear------计算y=ax+b

in_features为输入维度

out_features为输出维度

bias为b的值

torch.nn.Linear和torch.nn.MSELoss

2.torch.nn.MSELoss

size_average为是最后loss加起来过后是否除以N

reduce----因为loss为张量，reduce为true表示将所有的lossi累加达到降维的目的

torch.nn.Linear和torch.nn.MSELoss