From 61192ae572c70ecfe06fd233c07fec14b0285ced Mon Sep 17 00:00:00 2001 From: Oceania2018 Date: Sat, 5 Jun 2021 21:49:34 -0500 Subject: [PATCH] Fix ParseSlices for Tensor overload. --- src/TensorFlowNET.Core/Tensors/tensor_util.cs | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/TensorFlowNET.Core/Tensors/tensor_util.cs b/src/TensorFlowNET.Core/Tensors/tensor_util.cs index 110f38df..048106bb 100644 --- a/src/TensorFlowNET.Core/Tensors/tensor_util.cs +++ b/src/TensorFlowNET.Core/Tensors/tensor_util.cs @@ -680,7 +680,7 @@ would not be rank 1.", tensor.op.get_attr("axis"))); var end = new List(); var strides = new List(); - // 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));