Hi, I am getting the following assertion error for one hot when I pass reshape and reshape when I pass one hot. I’m writing
one_hot = tf.reshape(tf.one_hot(label, depth = C, axis = 0), shape = [C , 1])
Expected output shape is (4,)
but yours is (4, 1)
, because you are doing shape = [C , 1])
. Removing that 1
would resolve the issue.