四、pytorch的低阶API:1.张量的结构操作

  1. 创建张量
torch.tensor(list,dtype)
torch.arange(start,end,step)
torch.linespace(start,end,len)
torch.zeros(tuple)
torch.zeros_like(tensor)
torch.ones(tuple)
torch.ones_like(tensor)
torch.fill(tensor,value)#把tensor填充为全部是value
torch.rand()
torch.randn()
torch.normal()
torch.randperm(int)#整数随机排列
torch.eye(a,b)#单位阵
torch.diag(tensor)#对角阵
  1. 索引切片
#规则切片
tensor[int]#取行
tensor[a,b]=tensor[a][b]#取行列
tensor[a:b:c,d:e:f]#取多行多列
#不规则切片
torch.index_select(tensor,dim,index)
torch.masked_select(tensor,判断条件)#抽取符合条件的数
#修改元素
torch.where(判断,value1,value2)#符合条件修改为value1,不符合则为value2
torch.index_fill(tensor,dim,index,value)
torch.masked_fill(tensor,判断条件,value)
  1. 维度变换
torch.reshape(tensor,list)#改变形状
torch.squeeze(tensor)#减少维度
torch.unsqueeze(tensor,axis)#增加维度
torch.transpose(tensor,a,b)#交换维度
  1. 合并分割
torch.cat()#合并
torch.stack()#堆叠
torch.split(tensor,split_size_or_sections,dim)#切割
#torch中dim和axis混用
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值