pytorch数据类型与转换
torch定义了7种cpu tensor类型和8中gpu tensor类型
使⽤时，直接传⼊数字，就是按照形状初始化
torch.FloatTensor(2,3)
torch.DoubleTensor(2,3)
torch.ByteTensor(2,3)
torch.CharTensor(2,3)
torch.ShortTensor(2,3)
torch.IntTensor(2,3)
torch.LongTensor(2,3)
trunc函数类型注意在torch中对不同dtype的tensor原则上是不能混合运算的。
可以对tensor进⾏强⾏转换。
1  直接在tensor后⾯接.dtype()进⾏转换
import torch
a=torch.Tensor(3.1415)
a=a.float()
a=a.double()
a=a.int()
也可以对模型进⾏骚操作，即对模型的输出结果进⾏强制转化
self.lstm=nn.LSTM(input_b_dim,hidden_size=n_lstm_units,droput=1-keep_prob).double()
这样就直接将lstm的模型进⾏了强转化。但是不会影响梯度的backward
2  使⽤to转换
torch.lzeros(1,2).to(torch.double)
3 使⽤type()进⾏转化
可以设置获取默认tensor类型
torch.set_default_tensor_type(torch.double) _default_tensor_type()

版权声明：本站内容均来自互联网，仅供演示用，请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系QQ:729038198，我们将在24小时内删除。