From 7864db4790f2cd115f6656cbe5f86f99e02cef35 Mon Sep 17 00:00:00 2001 From: Sattisvar Tandabany Date: Thu, 10 Oct 2019 10:49:23 +0200 Subject: [PATCH] _compute_fans is buggy in VarianceScaling #420 --- .../Operations/Initializers/VarianceScaling.cs | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/TensorFlowNET.Core/Operations/Initializers/VarianceScaling.cs b/src/TensorFlowNET.Core/Operations/Initializers/VarianceScaling.cs index e2b2a0d6..e1ac0204 100644 --- a/src/TensorFlowNET.Core/Operations/Initializers/VarianceScaling.cs +++ b/src/TensorFlowNET.Core/Operations/Initializers/VarianceScaling.cs @@ -84,7 +84,7 @@ namespace Tensorflow.Operations.Initializers // Assuming convolution kernels (2D, 3D, or more). // kernel shape: (..., input_depth, depth) int receptive_field_size = 1; - foreach (var dim in shape.Take(2)) + foreach (var dim in shape.Take(shape.Length - 2)) receptive_field_size *= dim; var fan_in = shape[shape.Length - 2] * receptive_field_size; var fan_out = shape[shape.Length - 1] * receptive_field_size;