我正在努力打印特定图层的输出。请注意,我不希望输出类似<tf.Tensor 'Print_5:0' shape=(?, 1024) dtype=float32>
但将某些内容作为数组或数组列表。
我的主要代码:
from keras.models import Model
from keras.layers import Input, Dense
import tensorflow as tf
g = model.get_layer(name='dropout_2', index=None).output
tf.Print(g, [tf.shape(g)])
这不起作用,因为它给了我一个输出:
<tf.Tensor 'Print_5:0' shape=(?, 1024) dtype=float32>
我已经看到了与此问题相关的所有流行问题,但没有任何解决方法。 请帮忙。