Python заменяет часть данных в массиве 4D с двумерным массивом с использованием индексации

Question

Если я создаю массив np.zeros с именем «sparsity» формы (3,1,3,16), а затем создаю еще один массив с массивами формы (3,16) с именем «per_channel_sparsity».

Это правильный способ "заменить" каждую из 3 (3,16) матриц в 'sparsity' на матрицу per_channel_sparsity?

import numpy as np

sparsity = np.zeros((3,1,3,16)).astype(np.uint8)
per_channel_sparsity = np.random.rand((3,16)).astype(np.uint8)

for i in range(3):
    sparsity[i, 0, :, :] = per_channel_sparsity