AI Summary 60.12308.0 PyTorch torch.set_default_tensor_type 函数摘要 2026-08-03
PyTorch torch.set_default_tensor_type 函数 Pytorch torch 参考手册 torch.set_default_tensor_type 是 PyTorch 中用于设置默认张量类型的函数。 函数定义 torch.set_default_tensor_type(tensor_type) 使用示例 实例 [mycode4 type='python'] impo..

PyTorch torch.set_default_tensor_type 函数

Pytorch torch 参考手册 Pytorch torch 参考手册

torch.set_default_tensor_type 是 PyTorch 中用于设置默认张量类型的函数。

函数定义

torch.set_default_tensor_type(tensor_type)

使用示例

实例

import torch

# 设置默认张量类型为 double
torch.set_default_tensor_type(torch.DoubleTensor)

x = torch.zeros(2, 3)
print("默认 dtype:", x.dtype)

# 恢复默认
torch.set_default_tensor_type(torch.FloatTensor)

Pytorch torch 参考手册 Pytorch torch 参考手册