adding a note about torch.mean

mrdbourke / pytorch-deep-learning

Materials for the Learn PyTorch for Deep Learning: Zero to Mastery course.

https://learnpytorch.io

MIT License

10.49k stars 3.09k forks source link

Open usefulldivider opened 3 months ago

usefulldivider commented 3 months ago

dtype can be used to specify the type rather than using typecasting using type(torch.float32)