我正在用tensorflow
构建一个CNN
模型。我想从第一个卷积层 conv1 中的张量中提取矩阵。
我只能得到它的形状:Tensor("conv2d_24/Tanh:0", shape=(1, 4, 5, 1), dtype=float32)
我想提取输出张量的 (4,5( 矩阵。 如何在 Python 中执行此操作?
您可以通过运行以下命令从张量获取输出:
tensor_name.eval(feed_dict={...})
并用您第一次定义张量时称之为张量的任何内容替换tensor_name。
除非您正在寻找这一层的重量?在这种情况下,我认为这应该回答您的问题:在由 TensorFlow 训练的模型中获取一些权重的值