Browse Source

Fix ParseSlices for Tensor overload.

tags/v0.60-tf.numpy
Oceania2018 4 years ago
parent
commit
61192ae572
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      src/TensorFlowNET.Core/Tensors/tensor_util.cs

+ 2
- 2
src/TensorFlowNET.Core/Tensors/tensor_util.cs View File

@@ -680,7 +680,7 @@ would not be rank 1.", tensor.op.get_attr("axis")));
var end = new List<Tensor>();
var strides = new List<Tensor>();

// var index = 0;
var index = 0;
var (new_axis_mask, shrink_axis_mask) = (0, 0);
var (begin_mask, end_mask) = (0, 0);
var ellipsis_mask = 0;
@@ -692,7 +692,7 @@ would not be rank 1.", tensor.op.get_attr("axis")));
else
end.Add(stop);

// shrink_axis_mask |= (1 << index);
shrink_axis_mask |= (1 << index);

if (step == null)
strides.Add(tf.constant(1, dtype: start.dtype));


Loading…
Cancel
Save