Indexing Routines (mygrad.indexing_routines)# Generating index tensors# where(condition, [x, y]) Return elements chosen from x or y depending on condition.