Browse Source

fix rank_internal works for both kNN and CNN examples.

tags/v0.9
Oceania2018 6 years ago
parent
commit
06e6ae6fb3
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      src/TensorFlowNET.Core/Operations/array_ops.py.cs

+ 1
- 1
src/TensorFlowNET.Core/Operations/array_ops.py.cs View File

@@ -143,7 +143,7 @@ namespace Tensorflow
name = scope;
var input_tensor = ops.convert_to_tensor(input);
var input_shape = tensor_util.to_shape(input_tensor.shape);
if (optimize && input_shape.NDim > -1)
if (optimize && input_shape.NDim > 0)
return constant_op.constant(input_shape.NDim, dtype: tf.int32, name: name);
else
return gen_array_ops.rank(input, name);


Loading…
Cancel
Save