Torch 定义了九种CPU tensor类型和九种GPU tensor类型
Data type |
dtype |
CPU tensor |
GPU tensor |
---|---|---|---|
32位浮点型 |
|
|
|
64位浮点型 |
|
|
|
16位浮点型 |
|
|
|
8位整型 (无符号) |
|
|
|
8位整型 (有符号) |
|
|
|
16位整型 (有符号) |
|
|
|
32位整型 (有符号) |
|
|
|
64位整型 (有符号) |
|
|
|
布尔型 |
|
|
|
torch.Tensor
是默认的tensor类型 (torch.FloatTensor
)的别名。