tensornn.utils.flatten

tensornn.utils.flatten(inputs: Tensor) → Tensor

Flatten the inputs array. For example, a neural network cannot take in an image as input since it is 2D, so we can flatten it to make it 1D.