pytorch tensor索引中的none含义

None作为tensor的索引作用是增加维度

ac.shape

torch.Size([16])

　　

p = ac[None,None,None,None,:]
p.shape
 torch.Size([1, 1, 1, 1, 16]))

　　

posted @ 2022-04-04 20:34  vv_869  阅读(512)  评论(0)  编辑  收藏  举报