Spec-Zone.ru › TensorFlow 2.9

tf.compat.v1.estimator.experimental.dnn_logit_fn_builder

Функция-строитель для dnn logit_fn.

tf.compat.v1.estimator.experimental.dnn_logit_fn_builder(
    units,
    hidden_units,
    feature_columns,
    activation_fn,
    dropout,
    input_layer_partitioner,
    batch_norm
)
Аргументы
units Целое число, указывающее размерность слоя logit. В случае MultiHead это должно быть суммой размеров logit всех компонентных голов.
hidden_units Итерируемый набор целых чисел, количество скрытых узлов на каждом слое.
feature_columns Итерируемый набор feature_column._FeatureColumn входных данных модели.
activation_fn Функция активации, применяемая к каждому слою.
dropout Если не None, вероятность отбрасывания заданной координаты.
input_layer_partitioner Разделитель для входного слоя.
batch_norm Использовать ли пакетную нормализацию после каждого скрытого слоя.
Возвращает
Функция logit_fn (см. ниже).
Исключения
ValueError Если units не целое число.

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/compat/v1/estimator/experimental/dnn_logit_fn_builder

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API