目录

 【python】tensor数据转换为普通数据

在 PyTorch 中

示例：标量转换为 Python 数据

示例：多元素张量转换为 Python 列表

 【python】tensor数据转换为普通数据

如果你想将 Tensor 转换为普通的数据类型（例如 Python 数字或列表），可以使用以下方法：

在 PyTorch 中

你可以使用 .item() 或 .numpy() 方法：

• .item()：如果 Tensor 是标量（即只有一个元素），可以用 .item() 将其转换为 Python 数字。
• .numpy()：如果是一个多元素的 Tensor，可以将其转换为 NumPy 数组，再转换为普通的 Python 数据结构（如列表）。

示例：标量转换为 Python 数据

import torch

# 创建一个标量张量
tensor = torch.tensor(3.14)
print(type(tensor))  # <class 'torch.Tensor'>

# 将张量转换为普通 Python 数字
python_number = tensor.item()

print(type(python_number))  # <class 'float'>
print(python_number)  # 3.14

示例：多元素张量转换为 Python 列表

import torch

# 创建一个多元素的张量
tensor = torch.tensor([1, 2, 3, 4])
print(type(tensor))  # <class 'torch.Tensor'>

# 将张量转换为 NumPy 数组，再转换为 Python 列表
python_list = tensor.numpy().tolist()

print(type(python_list))  # <class 'list'>
print(python_list)  # [1, 2, 3, 4]