在PyTorch中,一些数据类型需要掌握,那么如何才能够自定义数据类型。接下来就是一些参考。
数据类型 |
dtype |
CPU张量 |
GPU张量 |
---|---|---|---|
32位浮点数(float) |
|
|
|
64位浮点数(double) |
|
|
|
16位浮点数(half) |
|
|
|
16位浮点数(bfloat16) |
|
|
在PyTorch中,一些数据类型需要掌握,那么如何才能够自定义数据类型。接下来就是一些参考。
数据类型 |
dtype |
CPU张量 |
GPU张量 |
---|---|---|---|
32位浮点数(float) |
|
|
|
64位浮点数(double) |
|
|
|
16位浮点数(half) |
|
|
|
16位浮点数(bfloat16) |
|
|