torch.set_default_tensor_type — PyTorch 2.6 documentation
pytorch.org