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)的别名。
2421

被折叠的 条评论
为什么被折叠?



