使用print_tensor在Keras中打印整个张量

Pab*_*blo 5 python keras tensorflow

我正在使用print_tensor来查看自定义损失函数中的张量值.但是,这只打印前三个值!像这样:

softmax = [[-0.245408952 -0.0407191925 -0.0813238621]...]

在Tensorflow中,您可以使用tf.Print中的summarize参数来控制它(在我的情况下不适用),但Keras的print_tensor没有这样的参数,那么我该如何改变这种行为呢?

到目前为止,我尝试过:

  1. 阅读print_tensor的文档(以及简要的代码),但我找不到任何参数
  2. 非常天真地设置numpy来编写整个数组

import numpy numpy.set_printoptions(threshold=numpy.nan)