golem ~master (2021-09-25T16:54:00Z)

BatchNorm.opCall

golem nn BatchNorm
Undocumented in source. Be warned that the author may not have intended to support it.
class BatchNorm(T, size_t[] Shape, UseGradient useGrad = UseGradient.yes)
auto
opCall
(
size_t[] ShapeX
UseGradient useGradX
)
(
Tensor!(T, ShapeX, useGradX) x
,
bool isTrain
)

Meta

Source

See Implementation

golem nn BatchNorm aliasesconstructorsfunctionsvariables

Page generated by adrdox