tf.equal(tf.argmax(y, 1), tf.argmax(labels, 1))
,
tf.equal(tf.argmax(y, -1), tf.argmax(labels, -1))
:
OP_REQUIRES(ctx, axis >= 0 && axis < input_dims,
errors::InvalidArgument("Expected dimension in the range [",
-input_dims, ", ", input_dims,
"), but got ", dim));