奇怪的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>

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

Answer 1

在这里找到答案

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

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