diff options
Diffstat (limited to 'tensorflow/contrib/labeled_tensor/python/ops/ops.py')
| -rw-r--r-- | tensorflow/contrib/labeled_tensor/python/ops/ops.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/tensorflow/contrib/labeled_tensor/python/ops/ops.py b/tensorflow/contrib/labeled_tensor/python/ops/ops.py index 590590bf7b..d846b013fe 100644 --- a/tensorflow/contrib/labeled_tensor/python/ops/ops.py +++ b/tensorflow/contrib/labeled_tensor/python/ops/ops.py @@ -727,10 +727,10 @@ def matmul(a, b, name=None): b = core.convert_to_labeled_tensor(b) if len(a.axes) > 2 or len(b.axes) > 2: - # We could use tf.batch_matmul to make this work, but we would also need - # to use tf.tile and/or tf.transpose. These are more expensive than doing - # reshapes, so it's not clear if it's a good idea to do this - # automatically. + # We could pass batched inputs to tf.matmul to make this work, but we + # would also need to use tf.tile and/or tf.transpose. These are more + # expensive than doing reshapes, so it's not clear if it's a good idea to + # do this automatically. raise NotImplementedError( 'matmul currently requires inputs with rank 2 or less, but ' 'inputs have ranks %r and %r' % (len(a.axes), len(b.axes))) |