数据类型

基本数据类型:
pytorch基础知识
练习:CPU和GPU下的数据类型
pytorch基础知识
pytorch基础知识


维度变换

dim 0、rank 0:注意tensor里面传的参数是1. 而不是[1.],二者有所区别,但是troch.tensor(1.).view(-1)之后返回的是dim为1。shape和size()都是返回tensor的每个维度的元素数目,但是shape是tensor的属性,size()是tensor的函数。
pytorch基础知识
pytorch基础知识
dim1、rank1:tensor 维度为1,传入创建tensor的是一个数组,调用from_numpy()可以从numpy对象中创建一个tensor对象,注意与dim0的区别,同时调用FloatTensor(2)返回的是维度为1的两个随机浮点数
pytorch基础知识
pytorch基础知识
dim 2:
pytorch基础知识
dim 3 :
pytorch基础知识
dim 4:
pytorch基础知识


索引与切片

Index:根据索引index获取元素

pytorch基础知识
select first/last N:

pytorch基础知识
select by steps:

pytorch基础知识
select by specific index:调用index_select函数实现

pytorch基础知识
省略号:只要选取的是1就可以把该维度去掉,只保留一个元素。

pytorch基础知识
select by mask:调用.masked_select()函数,返回的是一维数据。

pytorch基础知识
select by flatten index:先将tensor flatten成一维数据,再根据传入的tensor索引获取元素。

pytorch基础知识


tensor维度变换

view() / reshape(): 维度变换
pytorch基础知识
Squeeze v.s. unsqueeze :降低维度或者拓展维度
pytorch基础知识
pytorch基础知识

pytorch基础知识
squeeze:当维度元素维1时,才会发生降维操作

pytorch基础知识
Expand: broadcasting
pytorch基础知识
Repeat: memory copied
pytorch基础知识
.t : 仅限二维数据
pytorch基础知识
Transpose:
pytorch基础知识
permute:
pytorch基础知识

相关文章:

猜你喜欢
  • 2021-09-08
  • 2022-12-23
  • 2021-05-21
  • 2021-06-12
  • 2022-12-23
  • 2021-12-06
相关资源
相似解决方案