jupyter查看权值矩阵的函数

在 PyTorch 中，可以通过以下代码查看模型的权值矩阵：

import torch

# 加载模型
model = torch.load('model.pt')

# 获取第1层的权值矩阵
weights = model.layer1.weight.data

# 打印权值矩阵
print(weights)

其中，`'model.pt'` 是保存的模型文件路径，`layer1` 表示模型的第1层，`weight.data` 表示获取该层的权值矩阵。可以根据模型的结构和需要，修改对应的层数和参数名称来获取不同层的权值矩阵。