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)