torch中astype函数的用法

astype函数用于将张量的数据类型转换为指定的数据类型。它的语法如下：

torch.Tensor.astype(dtype)

其中，dtype是指定的数据类型，可以是torch.float、torch.double、torch.int等。

例如，将一个张量转换为float类型：

import torch

x = torch.tensor([1, 2, 3])
x = x.astype(torch.float)
print(x)

输出：

tensor([1., 2., 3.])