Fix an error message.
Minimum tensor rank should be (dim + 1) instead of (dim).
Example:
tf.argmin(x, 1) # x is a 1-d vector.
before:
"InvalidArgumentError: Minimum tensor rank: 1 but got: 1"
after:
"InvalidArgumentError: Minimum tensor rank: 2 but got: 1"
Loading
Please sign in to comment