数据类型
基本数据类型:
练习:CPU和GPU下的数据类型
维度变换
dim 0、rank 0:注意tensor里面传的参数是1. 而不是[1.],二者有所区别,但是troch.tensor(1.).view(-1)之后返回的是dim为1。shape和size()都是返回tensor的每个维度的元素数目,但是shape是tensor的属性,size()是tensor的函数。
dim1、rank1:tensor 维度为1,传入创建tensor的是一个数组,调用from_numpy()可以从numpy对象中创建一个tensor对象,注意与dim0的区别,同时调用FloatTensor(2)返回的是维度为1的两个随机浮点数
dim 2:
dim 3 :
dim 4:
索引与切片
Index:根据索引index获取元素
select first/last N:
select by steps:
select by specific index:调用index_select函数实现
省略号:只要选取的是1就可以把该维度去掉,只保留一个元素。
select by mask:调用.masked_select()函数,返回的是一维数据。
select by flatten index:先将tensor flatten成一维数据,再根据传入的tensor索引获取元素。
tensor维度变换
view() / reshape(): 维度变换
Squeeze v.s. unsqueeze :降低维度或者拓展维度
squeeze:当维度元素维1时,才会发生降维操作
Expand: broadcasting
Repeat: memory copied
.t : 仅限二维数据
Transpose:
permute: