python torch.int64转为int
时间: 2023-07-19 22:05:35 浏览: 221
基于torch.nn的回归、二分类、多分类任务代码
在 Python 中,`int` 类型是一种原生类型,而 `torch.int64` 是 PyTorch 中的数据类型。因此,将 `torch.int64` 转换为 `int` 类型需要使用 `item()` 方法,如下所示:
```python
import torch
# 创建一个 torch.int64 类型的张量
x = torch.tensor([1, 2, 3], dtype=torch.int64)
# 将 torch.int64 类型转换为 int 类型
y = x.item()
print(type(y)) # <class 'int'>
```
在上面的例子中,我们先创建了一个 `torch.int64` 类型的张量 `x`,然后使用 `item()` 方法将其转换为一个标量,最后将其赋值给变量 `y`,这样 `y` 就成为了一个 `int` 类型的整数。
阅读全文