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

Question

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

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

现在,我想转换y为Torch.LongTensor.我怎样才能做到这一点？

Answer 1

.to手电筒使用方法如下：

y = y.to(torch.long)

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

Answer 2

y = y:long()做的工作.存在用于其他数据类型,例如类似的方法int,char,float和byte

@OlivierRoche 这篇文章最初提到了 lua torch，其中 `y:long()` 是有效的语法。我想在过去几年里“torch”已经变成了“pytorch”的意思 (2认同)

Answer 3

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

y = y.type(torch.LongTensor)