tensor转list

在 PyTorch 中，可以使用 `.tolist()` 方法将 tensor 转换为 list。例如：

import torch

# 创建一个2x3的tensor
tensor = torch.tensor([[1, 2, 3], [4, 5, 6]])

# 将tensor转换为list
lst = tensor.tolist()

print(lst)  # [[1, 2, 3], [4, 5, 6]]

如果使用的是 TensorFlow，可以使用 `.numpy().tolist()` 方法将 tensor 转换为 list。例如：

import tensorflow as tf

# 创建一个2x3的tensor
tensor = tf.constant([[1, 2, 3], [4, 5, 6]])

# 将tensor转换为list
lst = tensor.numpy().tolist()

print(lst)  # [[1, 2, 3], [4, 5, 6]]

相关问题

tensor 转换list

将PyTorch中的list转换成tensor可以使用torch.Tensor()函数进行转换。例如：

import torch

a = [1, 2, 3]
tensor_a = torch.Tensor(a)
print(tensor_a)

输出结果为：

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

tensorflow tensor转list

你可以使用`tensor.numpy().tolist()`方法将 Tensor 转换为 Python 的 List。示例如下：

import tensorflow as tf

# 创建一个 Tensor
tensor = tf.constant([[1, 2], [3, 4]])

# 将 Tensor 转换为 List
lst = tensor.numpy().tolist()

print(lst)  # 输出：[[1, 2], [3, 4]]