我有形状为（N，M，D）的np个数组A。我想创建形状为（N，M，D，D）的np.ndarray B，使得对于沿着轴0和1的每对固定索引n，mB[n，m] =眼长（A[n，m]）我知道如何使用循环来解决这个问题，但是我想写代码来以矢量化的方式执行这个任务，使用numpy怎么做呢？
wz3gfoph1#
import numpy as np A = ... # Your array here n, m, d = A.shape indices = np.arange(d) B = np.zeros((n, m, d, d)) B[:, :, indices, indices] = A
1条答案
按热度按时间wz3gfoph1#