本文简要介绍python语言中 torch.set_default_tensor_type
的用法。
用法:
torch.set_default_tensor_type(t)
t(type或者string) -浮点张量类型或其名称
将默认的
torch.Tensor
类型设置为浮点张量类型t
。此类型也将用作torch.tensor()
中类型推断的默认浮点类型。默认浮点张量类型最初是
torch.FloatTensor
。例子:
>>> torch.tensor([1.2, 3]).dtype # initial default for floating point is torch.float32 torch.float32 >>> torch.set_default_tensor_type(torch.DoubleTensor) >>> torch.tensor([1.2, 3]).dtype # a new floating point tensor torch.float64
参数:
相关用法
- Python PyTorch set_default_dtype用法及代码示例
- Python PyTorch set_flush_denormal用法及代码示例
- Python PyTorch set_grad_enabled用法及代码示例
- Python PyTorch sentencepiece_numericalizer用法及代码示例
- Python PyTorch sentencepiece_tokenizer用法及代码示例
- Python PyTorch searchsorted用法及代码示例
- Python PyTorch saved_tensors_hooks用法及代码示例
- Python PyTorch sqrt用法及代码示例
- Python PyTorch skippable用法及代码示例
- Python PyTorch squeeze用法及代码示例
- Python PyTorch square用法及代码示例
- Python PyTorch save_on_cpu用法及代码示例
- Python PyTorch scatter_object_list用法及代码示例
- Python PyTorch skip_init用法及代码示例
- Python PyTorch simple_space_split用法及代码示例
- Python PyTorch sum用法及代码示例
- Python PyTorch sub用法及代码示例
- Python PyTorch sparse_csr_tensor用法及代码示例
- Python PyTorch symeig用法及代码示例
- Python PyTorch sinh用法及代码示例
- Python PyTorch sinc用法及代码示例
- Python PyTorch std_mean用法及代码示例
- Python PyTorch spectral_norm用法及代码示例
- Python PyTorch slogdet用法及代码示例
- Python PyTorch symbolic_trace用法及代码示例
注:本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torch.set_default_tensor_type。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。