tf.keras.layers.Softmax

Функция активации Softmax.

Наследуется от: Layer, Module

tf.keras.layers.Softmax(
    axis=-1, **kwargs
)

Пример без маски:

inp = np.asarray([1., 2., 1.])
layer = tf.keras.layers.Softmax()
layer(inp).numpy()
array([0.21194157, 0.5761169 , 0.21194157], dtype=float32)
mask = np.asarray([True, False, True], dtype=bool)
layer(inp, mask).numpy()
array([0.5, 0. , 0.5], dtype=float32)

Входная форма:

Произвольная. Используйте ключевой аргумент input_shape (кортеж целых чисел, не включая ось выборок) при использовании этого слоя в качестве первого слоя в модели.

Форма вывода:

Такая же, как у входных данных.

Аргументы вызова:

• inputs: Входные данные или логарифмы для слоя softmax.
• mask: Булевая маска такой же формы, как у inputs. По умолчанию None.