athena__initialiser_zeros Module

Module containing the implementation of the Zeros initialiser

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


Uses


Interfaces

public interface zeros_init_type

  • private module function initialiser_zeros_setup() result(initialiser)

    Interface for the Zeros initialiser

    Arguments

    None

    Return Value type(zeros_init_type)

    Zeros initialiser object


Derived Types

type, public, extends(base_init_type) ::  zeros_init_type

Type for the Zeros 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_zeros_setup ()

Interface for the Zeros initialiser

Type-Bound Procedures

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

Initialise the weights and biases using the Zeros distribution


Functions

private module function initialiser_zeros_setup() result(initialiser)

Interface for the Zeros initialiser

Arguments

None

Return Value type(zeros_init_type)

Zeros initialiser object


Subroutines

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

Initialise the weights and biases using the Zeros distribution

Arguments

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

Instance of the Zeros 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