ValueError: operands could not be broadcast together with shapes (480,20531) (20531,1)

[palanikumarraaj]

Step:
#normalizing with tensorflow numpy array
#normalize along vertical axis i.e. for each gene
train_feature_norm = tf.keras.utils.normalize(train_features, axis=0, order=2)

I have been trying your script for learning purpose. During the particular step, I got the above error for dimension issue.
Even so, I have used the same dataset and followed all the same steps so far.

It would be great, if you can explain that any step might missed in final script or how to tackle this error.
Note: I am beginner in ML and Python.

Thank you

PFA of the error for your reference.