pytorch中[...,0]的用法

Rootzzy1216

已于 2022-10-04 15:15:03 修改

阅读量484

点赞数

分类专栏： pytorch会用到的python知识文章标签： pytorch python

于 2022-10-04 14:54:23 首次发布

本文链接：https://blog.csdn.net/Rootzzy1216/article/details/127160628

版权

pytorch会用到的python知识专栏收录该内容

1 篇文章 0 订阅

订阅专栏

1、[...,0]是表示在哪一根轴上

对于轴的概念：(66条消息) 【Python 模块学习】NumPy中的维度(dimension)、轴(axis)、秩(rank)的含义_Babyfatliang的博客-CSDN博客_python秩

import torch
b=torch.Tensor([[[[10,2],[4,5],[7,8]],[[1,2],[4,5],[7,8]]]])
print(b.size())
torch.Size([1, 2, 3, 2])
c=b[...,0]
print(c)
tensor([[[10.,  4.,  7.],
         [ 1.,  4.,  7.]]])
d=b[...,1]
print(d)
tensor([[[2., 5., 8.],
         [2., 5., 8.]]])
e=b[:,:,:,0]
print(c==e)
tensor([[[True, True, True],
         [True, True, True]]])

参考：(66条消息) numpy数组中冒号[:,:,0]与[...,0]的区别_象驮着的云的博客-CSDN博客

numpy中的详细解释参考：(66条消息) Python中[-1]、[:-1]、[::-1]、[n::-1]、[:,:,0]、[…,0]、[…,::-1] 的理解_Dunkle.T的博客-CSDN博客_python[-1]