BatchNorm

Undocumented in source.

class BatchNorm (

T

size_t[] Shape

UseGradient useGrad = UseGradient.yes

) {

Tensor!(T, Shape, UseGradient.no) mean;

Tensor!(T, Shape, UseGradient.no) var;

Tensor!(T, Shape, useGrad) factor;

Tensor!(T, Shape, useGrad) offset;

Tensor!(T, Shape, UseGradient.no) tempMean;

Tensor!(T, Shape, UseGradient.no) tempVar;

Tensor!(T, Shape, UseGradient.no) temps;

alias parameters = AliasSeq!(mean, var, factor, offset);

auto opCall(Tensor!(T, ShapeX, useGradX) x, bool isTrain);

}

Constructors

this this(): Undocumented in source.

Members

Aliases

parameters alias parameters = AliasSeq!(mean, var, factor, offset): Undocumented in source.

Functions

opCall auto opCall(Tensor!(T, ShapeX, useGradX) x, bool isTrain): Undocumented in source. Be warned that the author may not have intended to support it.

Variables

factor Tensor!(T, Shape, useGrad) factor;: Undocumented in source.
mean Tensor!(T, Shape, UseGradient.no) mean;: Undocumented in source.
momentum T momentum;: Undocumented in source.
offset Tensor!(T, Shape, useGrad) offset;: Undocumented in source.
tempMean Tensor!(T, Shape, UseGradient.no) tempMean;: Undocumented in source.
tempVar Tensor!(T, Shape, UseGradient.no) tempVar;: Undocumented in source.
temps Tensor!(T, Shape, UseGradient.no) temps;: Undocumented in source.
var Tensor!(T, Shape, UseGradient.no) var;: Undocumented in source.

Meta

Source

See Implementation

golem nn classes

enums

hasParameters

structs

Activation