如何在张量流的自定义损失中获取张量的形状
how to get the shapes of a tensor inside a custom loss in tensorflow
我实施了自定义损失。我想获取输入参数的形状，例如 y_true 和 y_pred。但
无论我尝试过什么，我都无法获得有效的形状。我尝试的方法包括 y_true.shape、int_shape(shape)、y_true.get_shape returned (None, None).
你知道 TensorFlow 版本 <= 2.0 并不 运行 急切。因此，当您 运行 图表时，您将获得有效的形状。了解输出形状并以此为基础实现你的损失函数。 tf.shape(tensor) returns 一个有效的形状。
我实施了自定义损失。我想获取输入参数的形状，例如 y_true 和 y_pred。但 无论我尝试过什么，我都无法获得有效的形状。我尝试的方法包括 y_true.shape、int_shape(shape)、y_true.get_shape returned (None, None).
你知道 TensorFlow 版本 <= 2.0 并不 运行 急切。因此，当您 运行 图表时，您将获得有效的形状。了解输出形状并以此为基础实现你的损失函数。 tf.shape(tensor) returns 一个有效的形状。