Факел - Как изменить тип тензора?

Я создал перестановку чисел от 1 до 3.

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

Теперь я хочу преобразовать y в Torch.LongTensor. Как я могу это сделать?


torch lua
person una_dinosauria    schedule 10.12.2015    source источник

Ответы (3)


arrow_upward
34
arrow_downward

y = y.long() выполняет свою работу. Существуют аналогичные методы для других типов данных, таких как int, char, float и byte.

Вы можете проверить различные dtypes здесь.

person una_dinosauria    schedule 10.12.2015
comment
Там опечатка. Конечно, una_dinosauria означает y.long() - person Olivier Roche; 14.08.2019
comment
@OlivierRoche Этот пост изначально относился к lua ​​torch, где y:long() был допустимым синтаксисом. Я думаю, в последние несколько лет факел стал означать pytorch - person una_dinosauria; 20.12.2019

arrow_upward
9
arrow_downward

используйте .to метод горелки следующим образом:

y = y.to(torch.long)

Более подробную информацию о тензоре факела type/ops можно найти здесь.

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

person Rev_Nge    schedule 16.02.2021

arrow_upward
4
arrow_downward

Для пользователей pytorch, поскольку поиск типа тензора изменения в pytorch в google приводит к этой странице, вы можете сделать:

y = y.type(torch.LongTensor)
person Dharma    schedule 23.12.2020