Python 将K.tf.nn.max\u pool\u与带有3D输入张量的_argmax（）一起使用_Python_Keras_Max Pooling

Python 将K.tf.nn.max\u pool\u与带有3D输入张量的_argmax（）一起使用

python keras

Python 将K.tf.nn.max\u pool\u与带有3D输入张量的_argmax（）一起使用,python,keras,max-pooling,Python,Keras,Max Pooling,我想为3D图像实现SegNet（宽度、高度和深度，其中深度不是通道）。因此，在网络的解码器部分，我需要池索引。函数K.tf.nn.max_pool_和_argmax（）仅适用于二维图像（宽度和高度）。有一个函数maxpoolg3d，但它只返回池操作后的张量，而不返回索引有人知道这个问题的解决办法吗

我想为3D图像实现SegNet（宽度、高度和深度，其中深度不是通道）。因此，在网络的解码器部分，我需要池索引。函数K.tf.nn.max_pool_和_argmax（）仅适用于二维图像（宽度和高度）。有一个函数maxpoolg3d，但它只返回池操作后的张量，而不返回索引

有人知道这个问题的解决办法吗