I need to get the first and last dimension of numpy.ndarray of arbitrary size.
If I have, shape(A) = (3,4,4,4,4,4,4,3)
my first idea would be to do it result = shape(A)[0,-1], but that doesn't seem to work with tuples, why not?
Is there an easier way to do this than
s=shape(A)
result=(s[0], s[-1])
Thanks for any help
source
share