Torch - 如何更改张量类型？

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)

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

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

Answer 3

对于 pytorch 用户，因为在 google 中搜索 pytorch 中的更改张量类型会进入此页面，您可以执行以下操作：

y = y.type(torch.LongTensor)

Answer 4

奇怪的是，所有这些方法都将 float64 张量转换为 int64 张量，并截掉所有小数。这只是我的 pytorch 版本的问题还是其他问题？