torch.tensor.to

torch.tensor.to() 是一个 PyTorch 中的方法，用于将一个 Tensor 对象转换为指定的数据类型和设备。

语法如下：

to(device=None, dtype=None, non_blocking=False, copy=False)

参数说明：
- `device`：指定要移动到的设备，可以是 'cpu' 或 'cuda'。如果不指定，则不会改变 Tensor 对象所在的设备。
- `dtype`：指定要转换为的数据类型，可以是 torch 数据类型或 Python 数值类型。如果不指定，则不会改变 Tensor 对象的数据类型。
- `non_blocking`：如果为 True，则表示异步复制，即将 Tensor 对象移动到设备时不会阻塞主程序的执行。默认为 False。
- `copy`：如果为 True，则表示创建一个副本并将副本移动到指定的设备。默认为 False。

示例用法：

import torch

# 创建一个 CPU 上的 FloatTensor 对象
x = torch.tensor([1.0, 2.0, 3.0])

# 将 Tensor 对象移动到 GPU
x = x.to('cuda')

# 将 Tensor 对象转换为 DoubleTensor 类型
x = x.to(dtype=torch.double)

# 使用异步复制将 Tensor 对象移动到 GPU
x = x.to('cuda', non_blocking=True)

# 创建一个 GPU 上的 IntTensor 对象
y = torch.tensor([1, 2, 3], device='cuda', dtype=torch.int)

注意：在使用 to() 方法时，原来的 Tensor 对象并不会被修改，而是返回一个新的 Tensor 对象。因此，需要将转换后的结果赋值给一个新的变量或重新赋值给原来的变量。