ValueError: logits and labels must have the same shape ((None, 4, 255, 255, 4) vs (None, 4, 64, 64, 4)) tf2 code example

Example: ValueError: logits and labels must have the same shape ((None, 1) vs (None, 2))

# you should reshape your labels as 2d-tensor
# the first dimension will be the batch dimension and the second the scalar label)

y_train = np.asarray(train_labels).astype('float32').reshape((-1,1))
y_test = np.asarray(test_labels).astype('float32').reshape((-1,1))