deepboof.impl.forward.standard

Class SpatialBatchNorm_F32

java.lang.Object

deepboof.impl.forward.standard.BaseFunction<Tensor_F32>

deepboof.impl.forward.standard.FunctionBatchNorm_F32

deepboof.impl.forward.standard.SpatialBatchNorm_F32

All Implemented Interfaces:

BatchNorm, FunctionBatchNorm<Tensor_F32>, SpatialBatchNorm<Tensor_F32>, Function<Tensor_F32>

public class SpatialBatchNorm_F32
extends FunctionBatchNorm_F32
implements SpatialBatchNorm<Tensor_F32>

Implementation of SpatialBatchNorm for Tensor_F32

Field Summary

Fields inherited from class deepboof.impl.forward.standard.FunctionBatchNorm_F32

EPS, params, requiresGammaBeta

Fields inherited from class deepboof.impl.forward.standard.BaseFunction

miniBatchSize, parameters, shapeInput, shapeOutput, shapeParameters

Constructor Summary

Constructors

Constructor and Description
SpatialBatchNorm_F32(boolean requiresGammaBeta)

Method Summary

Modifier and Type	Method and Description
void	_forward(Tensor_F32 input, Tensor_F32 output)
void	_initialize()

Methods inherited from class deepboof.impl.forward.standard.FunctionBatchNorm_F32

_setParameters, getEPS, getTensorType, hasGammaBeta, setEPS

Methods inherited from class deepboof.impl.forward.standard.BaseFunction

forward, getOutputShape, getParameters, getParameterShapes, initialize, setParameters

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface deepboof.forward.SpatialBatchNorm

forward, setParameters

Methods inherited from interface deepboof.Function

getOutputShape, getParameters, getParameterShapes, getTensorType, initialize

Methods inherited from interface deepboof.forward.BatchNorm

getEPS, hasGammaBeta, setEPS

Methods inherited from interface deepboof.forward.FunctionBatchNorm

forward, setParameters

Constructor Detail

SpatialBatchNorm_F32

public SpatialBatchNorm_F32(boolean requiresGammaBeta)

Method Detail

_initialize

public void _initialize()

Overrides:

_initialize in class FunctionBatchNorm_F32

_forward

public void _forward(Tensor_F32 input,
                     Tensor_F32 output)

Overrides:

_forward in class FunctionBatchNorm_F32