【冰糖Python】PyTorch： Tensor 张量属性及相关操作

此处对张量（Tensor）的相关属性及计算方法做一个简单汇总

1、数据类型

16位浮点数：torch.float16 或 torch.half，torch.HalfTensor，torch.cuda.HalfTensor

32位浮点数：torch.float32 或 torch.float，torch.FloatTensor，torch.cuda.FloatTensor

64位浮点数：torch.float64 或 torch.double，torch.DoubleTensor，torch.cuda.DoubleTensor

8位无符号整型：torch.uint8，torch.ByteTensor，torch.cuda.ByteTensor

8位有符号整型：torch.int8，torch.CharTensor，torch.cuda.CharTensor

16位有符号整型：torch.int16 或 torch.short，torch.ShortTensor，torch.cuda.ShortTensor

32位有符号整型：torch.int32 或 torch.int，torch.IntTensor，torch.cuda.IntTensor

64位有符号整型：torch.int64 或 torch.long，torch.LongTensor，torch.cuda.LongTensor

pytorch 中默认的数据类型是 torch.float32

对于张量，可使用 .dtype 获取张量类型

2、生成张量

torch.tensor()：可将python列表或序列转换为张量，可使用dtype参数指定数据类型

torch.Tensor()：生成张量，可指定形状

同时，可以对 numpy 的 ndarray 和 tensor 相互转化，参见此文

torch.empty()：生成指定大小的空张量（随机填充）