pytorch学习3-基本数据类型

1.pytorch类型

pytorch学习3-基本数据类型
pytorch没有string类型,用one-hot,embedding表示string

2.数据类型

pytorch学习3-基本数据类型
常用 FloatTensor、IntTensor、ByteTensor

3.类型推断

torch.randn(2,3)  #初始化两行三列的tensor
a.type()          #查看类型
type(a)           #python自带方法,返回基本数据类型 torch.Tensor,没有额外信息
isinstance(a,torch.FloatTensor) #合法化检验 True/False

常用:a.type(),isinstance()

data = data.cuda()  #将CPU的数据data,搬运为GPU的数据

标量

例如1.0这样的数

a = torch.Tensor(1.)
a.dim()   #查看维度
len(a.shape)

loss用的最多就是维度为0的标量

上一篇:一位初学者对Mindspore的一些建议(1)


下一篇:听课笔记对应code(前四章)