将PyTorch张量转换为Python列表

Convert PyTorch tensor to python list(将PyTorch张量转换为Python列表)

本文介绍了将PyTorch张量转换为Python列表的处理方法,对大家解决问题具有一定的参考价值,需要的朋友们下面随着小编来一起学习吧!

问题描述

如何将PyTorchTensor转换为python列表?

我当前的用例是将大小[1, 2048, 1, 1]的张量转换为包含2048个元素的列表。

我的张量具有浮点值。是否有解决方案还可以解决INT和可能的其他数据类型?

推荐答案

使用Tensor.tolist()例如:

>>> import torch
>>> a = torch.randn(2, 2)
>>> a.tolist()
[[0.012766935862600803, 0.5415473580360413],
 [-0.08909505605697632, 0.7729271650314331]]
>>> a[0,0].tolist()
0.012766935862600803

若要删除大小1的所有维度,请使用a.squeeze().tolist()

或者,如果除一个维之外的所有维的大小均为1(或者您希望获取张量的每个元素的列表),则可以使用a.flatten().tolist()

这篇关于将PyTorch张量转换为Python列表的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持编程学习网!

本文标题为:将PyTorch张量转换为Python列表

基础教程推荐