yes the kth array is an index array so the indices can be negative which are then wrapped around the axis dimension like other numpy index arrays.