如何在张量流的自定义损失中获取张量的形状

Question

我实现了自定义损失。我想得到输入参数的形状，比如 y_true 和 y_pred。但 无论我尝试过什么，我都无法获得有效的形状。我尝试的方法包括 y_true.shape, int_shape(shape), y_true.get_shape 返回 (None, None)。

Answer 1

您知道 TensorFlow 版本 tf.shape(tensor) 返回一个有效的形状。