记:关于Pytorch中Linear结构与参数权重查看
记:关于Pytorch中Linear结构与参数权重查看
代码段:
import torch
x = torch.ones(1,3)
y = torch.nn.Linear(3,3,bias=True)
print(x)
print(y)
out = y.forward(x)
print(out)
print(y.state_dict().keys()) # 查看有哪些参量
print(y.weight) # 输出weight参量
print(y.bias)
原创
2021-10-21 20:18:59 ·
2882 阅读 ·
0 评论