public static Tensor BatchNorm(Tensor input, Tensor scale, Tensor offset)
{
var output = new Tensor(input.Shape);
GpuBackend.InvokeBatchNorm(input, scale, offset, output);
return(output);
}
// Batch normalization
public static void BatchNorm(Tensor input, Tensor scale, Tensor offset, Tensor output)
{
output.Reset(input.Shape);
GpuBackend.InvokeBatchNorm(input, scale, offset, output);
}
