Seite 1 von 1

axis Parameter von tf.keras.layers.BatchNormalization

Verfasst: Montag 3. Oktober 2022, 14:32
von nichtSoGuter
Um besser zu verstehen, was der "axis" Parameter von tf.keras.layers.BatchNormalization macht habe ich folgendes codebeispiel geschrieben. Ich habe den wert der axis auf axis=0; axis=1 und axis=2 gesetzt. Bei allen drei versuchen, war der output gleich. Daher erschließt sich mir der nutzen des Parameters nicht ganz.

Code: Alles auswählen

bn = tf.keras.layers.BatchNormalization(axis =2) # axis=0; axis=1

arr = np.arange(100, step = 10, dtype= np.float32)
arr = arr.reshape(2,-1,1)
output = bn(arr)
output
Kann mir jemand erklären, wieso das so ist?