commit e8dfd9c1d303773d3534ddc8eb0ceb2059b0d194
author fsx950223 <fsx950223@outlook.com> Wed Oct 23 16:18:25 2019 +0800
committer fsx950223 <fsx950223@outlook.com> Wed Nov 06 14:06:42 2019 +0800
tree a7b0b285693d2393a91c22d62b553f5345360fa7
parent 59df5f7dfa608d5a40ba4c07ba6a0954de1eeab7

fix strings.split

tensorflow/python/ops/ragged/ragged_string_ops.py

diff --git a/tensorflow/python/ops/ragged/ragged_string_ops.py b/tensorflow/python/ops/ragged/ragged_string_ops.py
index 5064f2f..2fe676b 100644
--- a/tensorflow/python/ops/ragged/ragged_string_ops.py
+++ b/tensorflow/python/ops/ragged/ragged_string_ops.py
@@ -22,6 +22,7 @@
 from tensorflow.python.framework import ops
 from tensorflow.python.ops import array_ops
 from tensorflow.python.ops import gen_string_ops
+from tensorflow.python.ops import gen_array_ops
 from tensorflow.python.ops import string_ops
 from tensorflow.python.ops.ragged import ragged_array_ops
 from tensorflow.python.ops.ragged import ragged_math_ops
@@ -630,6 +631,8 @@
     if result_type == "SparseTensor" and input.shape.rank == 1:
       return string_ops.string_split_v2(input, sep=sep, maxsplit=maxsplit)
 
+    if result_type == "SparseTensor" and input.shape.rank==0:
+      input = gen_array_ops.expand_dims(input, 0)
     ragged_result = string_split_v2(input, sep=sep, maxsplit=maxsplit)
     if result_type == "SparseTensor":
       return ragged_result.to_sparse()