nf_input3d_layer.f90

module nf_input3d_layer

  !! This module provides the `input3d_layer` type.

  use nf_base_layer, only: base_layer
  implicit none

  private
  public :: input3d_layer

  type, extends(base_layer) :: input3d_layer
    real, allocatable :: output(:,:,:)
  contains
    procedure :: init
    procedure :: set
  end type input3d_layer

  interface input3d_layer
    pure module function input3d_layer_cons(output_shape) result(res)
      !! Create a new instance of the 3-d input layer.
      !! Only used internally by the `layer % init` method.
      integer, intent(in) :: output_shape(3)
        !! Shape of the input layer
      type(input3d_layer) :: res
        !! 3-d input layer instance
    end function input3d_layer_cons
  end interface input3d_layer

  interface

    module subroutine init(self, input_shape)
      !! Only here to satisfy the language rules
      !! about deferred methods of abstract types.
      !! This method does nothing for this type and should not be called.
      class(input3d_layer), intent(in out) :: self
      integer, intent(in) :: input_shape(:)
    end subroutine init

    pure module subroutine set(self, values)
      class(input3d_layer), intent(in out) :: self
        !! Layer instance
      real, intent(in) :: values(:,:,:)
        !! Values to set
    end subroutine set

  end interface

end module nf_input3d_layer