Keras：如何从张量中仅提取某些层

Question

我有一个 4-D 张量形状[6,20,30,6]，我想执行 keras/tensorflow 等效于：new = np.array([old[i,:,:,i] for i in range(6)])任何帮助表示赞赏！

Answer 1

我能够在纯 keras 中进行转换。

def cc(x):
   return K.backend.stack([x[:,i, :, :, i] for i in range(6)], axis=1)

然后在 keras 模型定义中：

new=L.Lambda(lambda y: cc(y))(old)