Início / user-22783867

Samvel Safaryan's questions

Martin Hope
Samvel Safaryan
Asked: 2023-10-22 23:58:43 +0800 CST
  • 5

Deixe que tenhamos um tensor com forma $n\times d\times h\times w\times p\times p$ e queremos concatenar a matriz innet com forma $p\time p$, de modo que façamos uma matriz com forma $ n\vezes d\vezes ph\vezes pw$. Como eu posso fazer isso?

array([[[[[[ 0,  1,  2],
           [ 3,  4,  5],
           [ 6,  7,  8]],

          [[ 9, 10, 11],
           [12, 13, 14],
           [15, 16, 17]]],


         [[[18, 19, 20],
           [21, 22, 23],
           [24, 25, 26]],

          [[27, 28, 29],
           [30, 31, 32],
           [33, 34, 35]]]]]])

depois da concat

array([[[[0,   1,  2,  9, 10, 11],
         [3,   4,  5, 12, 13, 14],
         [6,   7,  8, 15, 16, 17],
         [18, 19, 20, 27, 28, 29],
         [21, 22, 23, 30, 31, 32],
         [24, 25, 26, 33, 34, 35]]]])

Fiz muitas experiências usando remodelar, mas sem sucesso. Um dos meus experimentos

a.reshape(n, d, p*h, p*w)

Posso fazer isso usando o loop for, mas acho que sem isso também é possível. Por favor me ajude. Código usando loop for

p = 3
arr = np.arange(1*1*2*2*p*p).reshape(1, 1, 2, 2, p, p)
answer = np.zeros(shape=(1, 1, 2*p, 2*p))

for (n, d, h, w) in np.ndindex(*arr.shape[:4]):
    answer[n, d, h:h+p, w:w+p] = arr[n, d, h, w]
numpy