本文介绍了PyTorch中Tensor的基础知识，包括Tensor作为多维矩阵的定义，以及如何从Python列表或数组转换为Tensor。重点讲解了torch.tensor()转换函数和tensor.item()方法，后者用于将单元素Tensor转换为Python数值。还提到了tolist()方法，用于将Tensor转换为Python列表。这些操作在PyTorch中非常常用。
最全面的官网说明：https://pytorch.org/docs/stable/tensors.html?highlight=item#
本文就平时用到的简单写一下，做备忘。
1. torch.Tensor就是一个多维矩阵，其元素均为一种数据类型。torch定义了9中CPU tensor类型和9中GPU tensor类型。具体看官网。
2. python 列表或数组->tensor 转换函数torch.tensor()
3. tensor的item()可以将tensor->python number
item() -> number
Returns the value of this tensor as a standard Python number. This only works for tensors with one element. For other cases, see tolist().
This operation is not differentiable.
Example:
>>> x = torch.tensor([1.0]) >>> x.item() 1.0
tolist() -> list or number
Returns the tensor as a (nested) list. For scalars, a standard Python number is returned, just
最低0.47元/天 解锁文章
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
