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

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

namespace Tensorflow.Keras.Layers
{
    public class GlobalAveragePooling1D : GlobalPooling1D
    {
        public GlobalAveragePooling1D(Pooling1DArgs 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_mean(inputs, 1, false);
            else
                return math_ops.reduce_mean(inputs, 2, false);
        }
    }
}