Browse Source

Fix reduced_shape #627

tags/v0.30
Oceania2018 5 years ago
parent
commit
dcb3f8b186
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      src/TensorFlowNET.Core/Operations/math_ops.cs

+ 2
- 2
src/TensorFlowNET.Core/Operations/math_ops.cs View File

@@ -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<int>())
input_shape_val[axes_val] = 1;
return tf.constant(input_shape_val);
}



Loading…
Cancel
Save