About the error of 'binary_rand_robust()' function in 'attack_feature_tf1.py'

[FanQiHang]

Hello! When the program executes the 'binary_rand_robust()' function in 'attack_feature_tf1.py', I change 'labels = np.argmax(ybatch, axis=-1)' to 'labels = ybatch.reshape(-1, len(ybatch))[0]' in this function according to my understanding of your paper. Should this be the case? If not, how should I modify it？ I‘m looking forward to your answer. Thank you.