From dcb3f8b18604c95f1d4d8f2c1a929a7c5dbd8a8e Mon Sep 17 00:00:00 2001 From: Oceania2018 Date: Sat, 24 Oct 2020 08:18:39 -0500 Subject: [PATCH] Fix reduced_shape #627 --- src/TensorFlowNET.Core/Operations/math_ops.cs | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/TensorFlowNET.Core/Operations/math_ops.cs b/src/TensorFlowNET.Core/Operations/math_ops.cs index 71faef7d..a3b638d3 100644 --- a/src/TensorFlowNET.Core/Operations/math_ops.cs +++ b/src/TensorFlowNET.Core/Operations/math_ops.cs @@ -436,8 +436,8 @@ namespace Tensorflow if(tf.Context.executing_eagerly()) { var input_shape_val = input_shape.numpy(); - var axes_val = (int)axes.numpy(); - input_shape_val[axes_val] = 1; + foreach (var axes_val in axes.numpy().ToArray()) + input_shape_val[axes_val] = 1; return tf.constant(input_shape_val); }