Keras 3 Custom Loss Function to mask NaN...

For keras layers with unknown sizes is it possible to know the output size for a specific input?...