Преобразование тензорного потока tf.contrib.layers.layer_norm в tf2.0

Question

Я хотел бы изменить следующий код с tf1.0 на tf2.0

 tf.contrib.layers.layer_norm(
      inputs=input_tensor, begin_norm_axis=-1, begin_params_axis=-1, scope=name)

Этот код взят из https://github.com/google-research/bert/blob/master/modeling.py Строка 364.

Пожалуйста, помогите мне.

Answer 1

Этот является эквивалентным способом сделать это в TF2.0

tf.keras.layers.LayerNormalization(input_tensor, axis = -1)