Thank you for making this consistent with the internal indexing by N x K x H x W.