在Python中,假设您有一个形状为零的2D数组(N,4,4):
z = array([[[0., 0., 0., 0.],
[0., 0., 0., 0.],
[0., 0., 0., 0.],
[0., 0., 0., 0.]],
[[0., 0., 0., 0.],
[0., 0., 0., 0.],
[0., 0., 0., 0.],
[0., 0., 0., 0.]]])
你有一个二维的索引数组:
i = array([[1, 1, 1, 1],
[1, 0, 0, 0],
[0, 0, 1, 1],