Torch - 如何更改张量类型？

Posted 2023-03-29

技术标签:

【中文标题】Torch - 如何更改张量类型？【英文标题】：Torch - How to change tensor type? 【发布时间】：2016-03-15 12:55:05 【问题描述】：

我创建了一个从 1 到 3 的数字排列。

th> y = torch.randperm(3 );
th> y
 3
 2
 1
[torch.DoubleTensor of size 3]

现在，我想将 y 转换为 Torch.LongTensor。我该怎么做？

【问题讨论】：

【参考方案1】：

y = y.long() 完成这项工作。其他数据类型也有类似的方法，如int、char、float和byte。

您可以检查不同的数据类型here。

【讨论】：

有一个错字。当然，una_dinosauria 的意思是 y.long() @OlivierRoche 这篇文章最初是指 lua torch，其中y:long() 是有效的语法。我猜在过去的几年里，“火炬”已经变成了“pytorch”的意思【参考方案2】：

使用torch的.to方法如下：

y = y.to(torch.long)

有关火炬张量类型/操作的更多详细信息可以在这里找到

https://pytorch.org/docs/stable/tensors.html

【讨论】：

【参考方案3】：

对于pytorch的用户，因为在google中搜索pytorch中的change tensor type会出现这个页面，所以可以这样做：

y = y.type(torch.LongTensor)

【讨论】：

以上是关于Torch - 如何更改张量类型？的主要内容，如果未能解决你的问题，请参考以下文章