I’m trying to build a simple three-layer neural network with tensorflow. When I pass a singular input into the model, I get a valid output in the form I want. However, when I try to fit the data in order to train it, I get the following error: