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