pytorch改变tensor的类型

假设变量名为ts

方法一：

ts=ts.double()
ts=ts.long()

方法二：

ts=ts.type(torch.DoubleTensor())
ts=ts.type(torch.LongTensor())

扫码关注

实用AI客栈

获取最新AI资讯与实战案例

小编微信号 : langu86