TensorFlow.NET/src/TensorFlowNET.Keras/Layers/Pooling/GlobalMaxPooling2D.cs

using System;
using System.Collections.Generic;
using System.Text;
using Tensorflow.Keras.ArgsDefinition;

namespace Tensorflow.Keras.Layers
{
    public class GlobalMaxPooling2D : GlobalPooling2D
    {
        public GlobalMaxPooling2D(Pooling2DArgs args)
            : base(args)
        {
        }

        protected override Tensors Call(Tensors inputs, Tensor mask = null, bool? training = null, Tensors initial_state = null, Tensors constants = null)
        {
            if (data_format == "channels_last")
                return math_ops.reduce_max(inputs, (1, 2), false);
            else
                return math_ops.reduce_max(inputs, (2, 3), false);
        }
    }
}