当前位置 : 主页 > 网络编程 > lua >

lua – 火炬 – 如何改变张量类型?

来源:互联网 收集:自由互联 发布时间:2021-06-23
我创建了从1到3的数字排列. th y = torch.randperm(3 );th y 3 2 1[torch.DoubleTensor of size 3] 现在,我想将y转换为Torch.LongTensor.我怎样才能做到这一点? y = y:long()完成工作.其他数据类型有类似的方法
我创建了从1到3的数字排列.

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

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

y = y:long()完成工作.其他数据类型有类似的方法,如int,char,float和byte
网友评论