PyTorch基础
一、基本数据类型-张量
Python与PyTorch中基本数据类型的对比
string在PyTorch中的表示方式有两种:
1.One-hot编码[0, 1, 0, 0, …],但该方式不能体现词与词之间的相关性
2.Embedding(略)
常用内建数据类型:
CPU | GPU |
---|---|
torch.FloatTensor | torch.cuda.FloatTensor |
torch.IntTensor | torch.cuda.IntTensor |
torch.ByteTensor | torch.cuda.ByteTensor |
类型检查
标量
维度检查
注:shape是数据成员;而size()是成员函数
维度为1的向量
注1:维度为1的向量常用于表示偏置(Bias)和线性层输入
注2:dim/rank和size/shape的区分:
dim表示维度;在数学中为rank
size是指tensor的具体形状
维度为2的向量
注:维度为2的向量常用于线性batch输入
维度为3的向量
注:维度为3的向量常用于RNN batch输入(RNN常用于NLP)
维度为4的向量
CNN:[batch, channel, height, width]
补充: