batchnorm Interface

interface
public module function batchnorm(input, params, momentum, mean, variance, epsilon) result(output)

Arguments

Type IntentOptional Attributes Name
class(array_type), intent(in), target :: input
class(array_type), intent(in), target :: params
real(kind=real32), intent(in) :: momentum
real(kind=real32), intent(in), dimension(:) :: mean
real(kind=real32), intent(in), dimension(:) :: variance
real(kind=real32), intent(in) :: epsilon

Return Value type(batchnorm_array_type),pointer