Torch-как изменить тип тензора?

Я создал перестановку чисел от 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