oa::FnMatrix::batchNorm2dBwd

batchNorm2dBwd: backward pass for BatchNorm2d.

FnC++ public headers

Function Documentation

BatchNorm2dBwdResult oa::FnMatrix::batchNorm2dBwd(
	const Matrix & inX,
	const Matrix & inGamma,
	const Matrix & inBeta,
	const Matrix & inMean,
	const Matrix & inVar,
	const Matrix & inOut,
	const Matrix & inDOut,
	oa::F32 inEps,
	bool inIsTraining
)

batchNorm2dBwd: backward pass for BatchNorm2d.

Parameters

inX

const Matrix &

—

inGamma

const Matrix &

—

inBeta

const Matrix &

—

inMean

const Matrix &

—

inVar

const Matrix &

—

inOut

const Matrix &

—

inDOut

const Matrix &

—

inEps

oa::F32

—

inIsTraining

bool

—

Returns

BatchNorm2dBwdResult

The declared return value.