【pyTorch基础】(1) 张量的数据类型

最新推荐文章于 2025-10-07 07:06:53 发布

原创

最新推荐文章于 2025-10-07 07:06:53 发布 · 2.3k 阅读

12 ·

CC 4.0 BY-SA版权

文章标签：

#深度学习 #python #pytorch #神经网络

最近想学一下图卷积GCN，教程竟然都是pyTorch的，作为一名重度TensorFlow用户太崩溃了，没办法只能学一波pyTorch了。

首先在pyTorch中，一个张量（tensor）可能会被部署在CPU上，也可能被部署在GPU上。它们的数据类型是不同的。即使是同一个数据，被部署在不同位置，那么它的数据类型也是不一样的。数据类型如下：

import torch
'''
# pyTorch的数据类型
# -------------------------------------- #
#（1）CUP上的数据类型
# torch.FloatTensor       (float32)
# torch.DoubleTensor      (float64)
# torch.ByteTensor        (float16)
# torch.CharTensor        (uint8)
# torch.ShortTensor       (int8)
# torch.InitTensor        (int16)
# torch.LongTensor        (int64)
# -------------------------------------- #
#（2）GUP上的数据类型
# torch.cuda.FloatTensor       (float32)
# torch.cuda.DoubleTensor      (float64)
# torch.cuda.ByteTensor        (float16)
# torch.cuda.CharTensor        (uint8)
# torch.cuda.ShortTensor       (int8)
# torch.cuda.InitTensor        (int16)
# torch.cuda.LongTensor        (int64)
# -------------------------------------- #
'''