torch.Tensor.bfloat16 — PyTorch 1.10.1 documentation
pytorch.org › docs › stableLearn about PyTorch’s features and capabilities. Community. Join the PyTorch developer community to contribute, learn, and get your questions answered. Developer Resources. Find resources and get questions answered. Forums. A place to discuss PyTorch code, issues, install, research. Models (Beta) Discover, publish, and reuse pre-trained models
Automatic Mixed Precision package - PyTorch
https://pytorch.org/docs/stable/amp.htmlAutomatic Mixed Precision package - torch.cuda.amp¶. torch.cuda.amp and torch provide convenience methods for mixed precision, where some operations use the torch.float32 (float) datatype and other operations use torch.float16 (half).Some ops, like linear layers and convolutions, are much faster in float16.Other ops, like reductions, often require the dynamic …