python - 奇怪的 tf.Print 错误

Question

我正在尝试使用这样的 tf.Print:

residual = tf.Print(residual, [residual], message='enc', summarize=100)

但它因以下错误而崩溃:

ValueError: Single tensor passed to 'data', expected list while building NodeDef 'tf_op_layer_tf_op_layer_TransformerEncoder/TransformerEncoderBlock/Print/TransformerEncoder/TransformerEncoderBlock/Print' using Op<name=Print; signature=input:T, data: -> output:T; attr=T:type; attr=U:list(type),min=0; attr=message:string,default=""; attr=first_n:int,default=-1; attr=summarize:int,default=3; is_stateful=true>

这对我来说毫无意义，因为数据参数包含在列表中。

Answer 1

在这里找到答案:

https://epcsirmaz.blogspot.com/2018/06/display-full-value-of-tensor-in.html

基本上，使用 Keras 时，您必须将其包装在 lambda 层中。