Я пытаюсь установить точную c диагональ трехмерной матрицы в ноль, в куб. Код выглядит следующим образом:
arr = cp.random.normal(0, 1, (100, 100, 100))
r = cp.arange(100, dtype=int)
arr[:, r, r] = 0
Чтобы сократить время, я надеюсь перенести этот последний шаг в ядро ElementWise. Но как мне указать индекс в 3d матрице?