diff --git a/tensorflow/python/ops/array_ops.py b/tensorflow/python/ops/array_ops.py index d15010608ba..e958cdfa613 100644 --- a/tensorflow/python/ops/array_ops.py +++ b/tensorflow/python/ops/array_ops.py @@ -1288,7 +1288,7 @@ def split(value, num_or_size_splits, axis=0, num=None, name="split"): tf.shape(split1) ==> [5, 15] tf.shape(split2) ==> [5, 11] # Split 'value' into 3 tensors along dimension 1 - split0, split1, split2 = tf.split(value=1, num_or_size_splits=3, axis=value) + split0, split1, split2 = tf.split(value, num_or_size_splits=3, axis=1) tf.shape(split0) ==> [5, 10] ```