我需要获取任意大小的 numpy.ndarray 的第一个和最后一个维度。
如果我的数组形状为
有没有比这更简洁的方法?
如果我的数组形状为
shape(A) = (3,4,4,4,4,4,4,3)
,我第一个想法是执行 result = shape(A)[0,-1]
,但似乎这在元组中不起作用,为什么?有没有比这更简洁的方法?
s=shape(A)
result=(s[0], s[-1])
感谢任何帮助