テラス軸をスラッシュで入れ替えます

Question

イメージバッチのテンソル軸を（batch_size、row、col、ch）から （batch_size、ch、row、col）にスワップします。 numpyの中

、これは私がKerasでそれを行うだろうか

X_batch = np.moveaxis(X_batch, 3, 1) 

で行うことができますか？

Answer 1

K.permute_dimensions()は、np.transpose()とまったく同じです。

例：

import numpy as np 
from keras import backend as K 

A = np.random.random((1000,32,64,3)) 
# B = np.moveaxis(A, 3, 1) 
C = np.transpose(A, (0,3,1,2)) 

print A.shape 
print C.shape 

A_t = K.variable(A) 
C_t = K.permute_dimensions(A_t, (0,3,1,2)) 

print K.eval(A_t).shape 
print K.eval(C_t).shape