PyTorch的数据类型与创建
数据对比:
python | PyTorch |
---|---|
Int | IntTensor of size() |
Float | FloatTensor of size() |
Int array | IntTensor of size[d1, d2,…] |
Float | FloatTensor of size [d1, d2,…] |
string | 没有对string的支持,可以One-hot/Embedding的编码表示 |
数据类型
Data type | dtype | CPU tensor | GPU tensor |
---|---|---|---|
32-bit floating point | torch.float32 or torch.float | torch.FloatTensor | torch.cuda.FloatTensor |
64-bit floating point | torch.float64 or torch.doble | torch.DoubleTensor | to |