diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-09-25 21:54:20 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-25 21:58:08 -0700 |
commit | f2b17b22e12bd743b66945070f338f70b5fa3332 (patch) | |
tree | b0d78ae134c6e241547970adedd078574a471920 /tensorflow/core/framework/tensor_test.cc | |
parent | 6666516f390f125ed70ddbd4e6f89b83d953c408 (diff) |
Allow subslicing Tensors with a single dimension.
PiperOrigin-RevId: 214553359
Diffstat (limited to 'tensorflow/core/framework/tensor_test.cc')
-rw-r--r-- | tensorflow/core/framework/tensor_test.cc | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/core/framework/tensor_test.cc b/tensorflow/core/framework/tensor_test.cc index 0bfa53e6c5..c596604143 100644 --- a/tensorflow/core/framework/tensor_test.cc +++ b/tensorflow/core/framework/tensor_test.cc @@ -1246,6 +1246,9 @@ TEST(Tensor, SubSlice_Basic) { EXPECT_EQ(&tx(5, j, k), &ty(j, k)); } } + Tensor z = y.SubSlice(3).SubSlice(31); + auto tz = z.unaligned_flat<float>(); + EXPECT_EQ(*tz.data(), 5.0); } { // Test unaligned access via a SubSlice. |