tf.print 不在控制台中打印打印件

Question

我有以下损失 function 包含 tf.print 语句：

@tf.function
def loss(i, o):
  tf.print("i shape is ")
  tf.print(i.shape)
  tf.print("o shape is ")
  tf.print(o.shape)
  loss = tf.math.reduce_sum(tf.abs(i - o))
  tf.print("loss is ")
  tf.print(loss)
  tf.print(loss)
  return loss

并且损失在训练步骤中被称为如下：

l_loss = loss(l_output, x_train_left_noc)

但是，控制台中没有任何打印。 怎样看印记？

Answer 1

import tensorflow as tf
import sys

tensor = tf.range(10)
tf.print(tensor, output_stream=sys.stderr)

我正在使用 tensorflow 2.x
指定output_stream=sys.stderr