张量操作

a = torch.randn(1,1,3,1)

a.expand(2,1,3,1)
a.expand(1,2,3,1)
a.expand(1,1,6,1)#runtimerror
a.expand(1,1,3,2)
a.expand(1,1,3,1)#a.expand(1,1,3,1) == a

b = torch.randn(1,1,2,2)

b.repeat(1,1,1,1)#b.repeat(1,1,1,1) == b
b.repeat(1,1,1,2)#b.repeat(1,1,1,2).shape -> torch.Size([1,1,2,4])
b.repeat(1,3,1,2)#b.repeat(1,3,1,2).shape -> torch.Size([1,3,2,4])

b.repeat(1,3,1,1) == b.expand(1,3,2,2) #return torch.ones(1,3,2,2,dtype=torch.uint8) or torch.ones([1,3,2,2],dtype=uint8)

'''view()'''
b.view(4)#torch.Size([4])
b.view(1,1,2,2)#b.view(1,1,2,2) == b
b.view(1,2,1,-1)#torch.Size([1,2,1,2])

'''torch.cat([],1)'''
c = torch.randn(1,2,2,2)

torch.cat([b,c],0)#RuntimeError
torch.cat([b,c],1)#torch.Size([1,3,2,2])除要合并的通道外,其余通道必须保持一致
torch.cat([b,c],2)#RuntimeError
torch.cat([b,c],3)#RuntimeError

'''torch.sqeeuze()''&#
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值