* Allow nagative values for batch_dims * Update formula * Update spec according to comments * clarified cases when batch_dims and axis less than zero and enhanced restriction for index types Co-authored-by: Pavel Esir <pavel.esir@intel.com>