NN_rms_norm()

void NN_rms_norm	(	Tensor *	y,
		const Tensor *	x,
		const Tensor *	w,
		float	eps
	)

Computes the root mean square normalization of a tensor.

Parameters

x	the input tensor of shape (m, n)
y	the output tensor of shape (m, n)