Task: Numpy Slice of Arbitrary Dimensions

Task: Numpy Slice of Arbitrary Dimensions: a task in Terminal-Lego-15k (Harbor dataset). You need to write a Python function that can slice a numpy array to obtain the i -th index in the last dimension , regardless of how many dimensions the array has.