what nn.flatten() does?
Flattens out all the dimensions to single dimensions. Consider you have 3x5x5, when you flatten it will be 75.
Ok understood
thank you!
what nn.flatten() does?
Flattens out all the dimensions to single dimensions. Consider you have 3x5x5, when you flatten it will be 75.
Ok understood
thank you!