r - 如何提取在卷积层中学习的过滤器？

Question

我的 R 代码中有一个卷积层，创建为：
conv1 <- mx.symbol.Convolution(data=data, kernel=c(10,1), num_filter=10)

网络完全训练后，如何提取 10 个过滤器？

Answer 1

过滤器权weight重在卷积的参数中。假设您使用了标准布局，如您的示例所示，权重将具有形状（num_filter、channels、kernel[0]、kernel[1]）。

例如

conv1.weight.data()[0]

从当前上下文访问第一个过滤器的权重张量。