matvec added to math_ops.py. This is pure sugar:
matvec(a, b) = matmul(a, b[..., tf.newaxis])[..., 0] PiperOrigin-RevId: 220198397
Loading
Please sign in to comment
matvec(a, b) = matmul(a, b[..., tf.newaxis])[..., 0] PiperOrigin-RevId: 220198397