Investigate if the Layer Output Shape can be removed from the user implementation and inferred by running the model with a dummy tensor
Investigate if the Layer Output Shape can be removed from the user implementation and inferred by running the model with a dummy tensor