标签： pytorch convolutional-neural-network
我希望打开/关闭CNN上各层中的一些过滤器。 我想要的是下面。 在测试期间，我将批量大小设置为1.然后，第一批进入网络。 网络输出softmax值。 使用此softmax值，我想关闭网络中的过滤器，这与低softmax值密切相关。 然后下一批，关闭过滤器不执行计算，并且softmax结果的数量少于第一批。
我可以使用pytorch吗？