此处对张量(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():生成指定大小的空张量(随机填充)