athena__initialiser_ones Module

Module containing the implementation of the Ones initialiser

This module contains the implementation of the Ones initialiser for the weights and biases of a layer


Uses


Interfaces

public interface ones_init_type

  • private module function initialiser_ones_setup() result(initialiser)

    Interface for the Ones initialiser

    Arguments

    None

    Return Value type(ones_init_type)

    Ones initialiser object


Derived Types

type, public, extends(base_init_type) ::  ones_init_type

Type for the Ones initialiser

Components

Type Visibility Attributes Name Initial
real(kind=real32), public :: mean = 1._real32

Scale, mean, and standard deviation of the initialiser

character(len=20), public :: name

Name of the initialiser

real(kind=real32), public :: scale = 1._real32

Scale, mean, and standard deviation of the initialiser

real(kind=real32), public :: std = 0.01_real32

Scale, mean, and standard deviation of the initialiser

Constructor

private module function initialiser_ones_setup ()

Interface for the Ones initialiser

Type-Bound Procedures

procedure, public, pass(this) :: initialise => ones_initialise

Initialise the weights and biases using the Ones distribution


Functions

private module function initialiser_ones_setup() result(initialiser)

Interface for the Ones initialiser

Arguments

None

Return Value type(ones_init_type)

Ones initialiser object


Subroutines

private pure subroutine ones_initialise(this, input, fan_in, fan_out, spacing)

Initialise the weights and biases using the Ones distribution

Arguments

Type IntentOptional Attributes Name
class(ones_init_type), intent(inout) :: this

Instance of the Ones initialiser

real(kind=real32), intent(out), dimension(..) :: input

Weights and biases to initialise

integer, intent(in), optional :: fan_in

Number of input and output parameters

integer, intent(in), optional :: fan_out

Number of input and output parameters

integer, intent(in), optional, dimension(:) :: spacing

Spacing of the input and output units