tf.raw_ops.XlaSparseCoreAdagradMomentum
tf.raw_ops.XlaSparseCoreAdagradMomentum(
indices,
gradient,
learning_rate,
beta_1,
epsilon,
accumulator,
momentum,
embedding_table,
feature_width,
use_nesterov,
beta_2,
exponent,
name=None
)
| Аргументы |
indices | A Tensor типа int32. |
gradient | A Tensor типа float32. |
learning_rate | A Tensor типа float32. |
beta_1 | A Tensor типа float32. |
epsilon | A Tensor типа float32. |
accumulator | A Tensor типа float32. |
momentum | A Tensor типа float32. |
embedding_table | A Tensor типа float32. |
feature_width | Целое число. |
use_nesterov | Строка. |
beta_2 | Логическое значение. |
exponent | Объект. |
name | Название операции (необязательно). |
| Возвращаемое значение |
Кортеж объектов Tensor (updated_embedding_table, updated_accumulator, updated_momentum). |
updated_embedding_table | A Tensor типа float32. |
updated_accumulator | A Tensor типа float32. |
updated_momentum | A Tensor типа float32. |