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); }