BatchNormalization(axis=3, epsilon=1e-05, momentum=0.1)
I want to convert this tensoflow code to pytorch, but pytorch does not have axis, how should I convert it?
Integer, the axis to be normalized (usually the feature axis). For example, set axis=1 in BatchNormalization after a Conv2D layer with data_format=“channels_first”. However, pytorch does not allow you to specify an axis.