火炬 - 如何改变张量类型？

Question

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

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

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

Answer 1

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

Answer 2

使用 .to 火炬方法如下：

y = y.to(torch.long)

可以在此处找到有关火炬张量 type/ops 的更多详细信息

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

Answer 3

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

y = y.type(torch.LongTensor)