pytorch数据变为long
时间: 2023-11-17 09:04:17 浏览: 116
将pytorch转成longtensor的简单方法
5星 · 资源好评率100%
要将PyTorch中的数据变为long类型,你可以使用`to()`方法将其转换为long类型。以下是示例代码:
```python
import torch
# 创建一个float类型的Tensor
data = torch.tensor([1.0, 2.0, 3.0])
# 将数据类型转换为long
data_long = data.to(torch.long)
print(data_long)
```
在上述示例中,我们首先创建了一个包含float类型数据的Tensor `data`。然后,我们使用`to()`方法将其转换为long类型,并将结果存储在`data_long`变量中。最后,我们打印出了转换后的结果。
请注意,`to()`方法还可以用于将数据移动到其他设备上,例如GPU。在这种情况下,你可以传递`device`参数来指定目标设备。例如,`data.to(torch.long, device='cuda')`将数据移动到CUDA设备上并将其转换为long类型。
阅读全文