📌  相关文章
📜  使用 pytorch 可视化卷积核 - Python 代码示例

📅  最后修改于: 2022-03-11 14:45:40.027000             🧑  作者: Mango

代码示例1
conv1 = nn.Conv2d(3, 1, 3)
weight = conv1.weight.data.numpy()
plt.imshow(weight[0, ...])