目录
一、Tensor的基本定义
张量(tensor)是pytorch中最基础的一种数据结构 ,在pytorch中的操作都是基于tensor。我们可能对标量,向量,矩阵都非常熟悉,其实张量是标量、矩阵等更为高维、更为普遍的表现形式。
二、Tensor的类型
在 torch 中 分为CUP 存储运算的tensor和 GPU存储运算的张量。Tensor在cpu、gpu下分别有 8 种数据类型,其中默认的数据类型为 32 位浮点型
数据类型 | dtype | CPU Tensor | GPU Tensor |
16 位浮点型 | torch.float16 或 torch.half | torch.HalfTensor | torch.cuda.HalfTensor |
32 位浮点型 | torch.float32 或 torch.float | torch.FloatTensor | torch.cuda.FloatTensor |
64 |