Commit 6c40bb5e authored by Ian Langmore's avatar Ian Langmore Committed by TensorFlower Gardener
Browse files

tf.linalg.matvec: Use tf.squeeze and expand_dims rather than implicit [..., 

newaxis] and [..., 0].  I was mistaken in thinking the latter had better
performance.  There is no verified difference.

PiperOrigin-RevId: 220553967
parent 2f999200
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment