diff options
Diffstat (limited to 'tensorflow/python/ops/math_ops.py')
| -rw-r--r-- | tensorflow/python/ops/math_ops.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/tensorflow/python/ops/math_ops.py b/tensorflow/python/ops/math_ops.py index 20b9e438a1..0dafb5d78b 100644 --- a/tensorflow/python/ops/math_ops.py +++ b/tensorflow/python/ops/math_ops.py @@ -1673,13 +1673,15 @@ def matmul(a, # 3-D tensor `a` - a = tf.constant(np.arange(1,13), shape=[2, 2, 3]) => [[[ 1. 2. 3.] + a = tf.constant(np.arange(1, 13, dtype=np.int32), + shape=[2, 2, 3]) => [[[ 1. 2. 3.] [ 4. 5. 6.]], [[ 7. 8. 9.] [10. 11. 12.]]] # 3-D tensor `b` - b = tf.constant(np.arange(13,25), shape=[2, 3, 2]) => [[[13. 14.] + b = tf.constant(np.arange(13, 25, dtype=np.int32), + shape=[2, 3, 2]) => [[[13. 14.] [15. 16.] [17. 18.]], [[19. 20.] |