tf.raw_ops.XlaSparseDenseMatmulGradWithFtrlAndCsrInput

Просмотр псевдонимов

Псевдонимы совместимости для миграции

См. Руководство по миграции для получения дополнительной информации.

tf.compat.v1.raw_ops.XlaSparseDenseMatmulGradWithFtrlAndCsrInput

tf.raw_ops.XlaSparseDenseMatmulGradWithFtrlAndCsrInput(
    row_pointers,
    sorted_sample_ids,
    sorted_token_ids,
    sorted_gains,
    activation_gradients,
    learning_rate,
    embedding_table,
    accumulator,
    linear,
    num_minibatches_per_physical_sparse_core,
    multiply_linear_by_learning_rate,
    beta,
    learning_rate_power,
    l1_regularization_strength,
    l2_regularization_strength,
    table_name,
    clip_weight_min=float('-inf'),
    clip_weight_max=float('inf'),
    name=None
)

Аргументы
`row_pointers`	A `Tensor` типа `int32`.
`sorted_sample_ids`	A `Tensor` типа `int32`.
`sorted_token_ids`	A `Tensor` типа `int32`.
`sorted_gains`	A `Tensor` типа `float32`.
`activation_gradients`	A `Tensor` типа `float32`.
`learning_rate`	A `Tensor` типа `float32`.
`embedding_table`	A `Tensor` типа `float32`.
`accumulator`	A `Tensor` типа `float32`.
`linear`	A `Tensor` типа `float32`.
`num_minibatches_per_physical_sparse_core`	A `Tensor` типа `int32`.
`multiply_linear_by_learning_rate`	A `bool`.
`beta`	A `float`.
`learning_rate_power`	A `float`.
`l1_regularization_strength`	A `float`.
`l2_regularization_strength`	A `float`.
`table_name`	A `string`.
`clip_weight_min`	Необязательный `float`. По умолчанию `float('-inf')`.
`clip_weight_max`	Необязательный `float`. По умолчанию `float('inf')`.
`name`	Имя операции (необязательно).

Возвращаемые значения
Кортеж из объектов `Tensor` (updated_embedding_table, updated_accumulator, updated_linear).
`updated_embedding_table`	A `Tensor` типа `float32`.
`updated_accumulator`	A `Tensor` типа `float32`.
`updated_linear`	A `Tensor` типа `float32`.

© 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/api_docs/python/tf/raw_ops/XlaSparseDenseMatmulGradWithFtrlAndCsrInput