init_gradients_rmsprop Subroutine

  pure subroutine init_gradients_rmsprop(this, num_params)
    !! Initialise gradients for RMSprop optimiser
    implicit none

    ! Arguments
    class(rmsprop_optimiser_type), intent(inout) :: this
    !! Instance of the RMSprop optimiser
    integer, intent(in) :: num_params
    !! Number of parameters


    ! Initialise gradients
    if(allocated(this%moving_avg)) deallocate(this%moving_avg)
    allocate(this%moving_avg(num_params), source=0._real32)
  end subroutine init_gradients_rmsprop