The initial tensor shapes should be unknown. Currently, they are set to the MNIST dataset shape. This makes the current tests pass.
The initial tensor shapes should be unknown. Currently, they are set to the MNIST dataset shape. This makes the current tests pass.