Skip to content

Commit 93d2bdc

Browse files
Corrected normalize() signature
1 parent 8792413 commit 93d2bdc

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/TorchSharp/NN/Normalization/Functional.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ public static Tensor normalize(Tensor input, double p = 2.0, long dim = 1L, doub
3131
/// <summary>
3232
/// Applies Batch Normalization for each channel across a batch of data.
3333
/// </summary>
34-
public static Tensor batch_norm(Tensor input, Tensor running_mean, Tensor running_var, Tensor? weight = null, Tensor? bias = null, bool training = false, double momentum = 0.1, double eps = 1e-5)
34+
public static Tensor batch_norm(Tensor input, Tensor? running_mean, Tensor? running_var, Tensor? weight = null, Tensor? bias = null, bool training = false, double momentum = 0.1, double eps = 1e-5)
3535
{
3636
var res = THSNN_batch_norm(
3737
input.Handle,

0 commit comments

Comments
 (0)