tf.raw_ops.XlaSparseDenseMatmulGradWithAdagradAndCsrInput
tf.raw_ops.XlaSparseDenseMatmulGradWithAdagradAndCsrInput(
row_pointers,
sorted_sample_ids,
sorted_token_ids,
sorted_gains,
activation_gradients,
learning_rate,
embedding_table,
accumulator,
num_minibatches_per_physical_sparse_core,
table_name,
clip_weight_min=float('-inf'),
clip_weight_max=float('inf'),
name=None
)
| Аргументы |
row_pointers | Tensor типа int32. |
sorted_sample_ids | Tensor типа int32. |
sorted_token_ids | Tensor типа int32. |
sorted_gains | Tensor типа float32. |
activation_gradients | Tensor типа float32. |
learning_rate | Tensor типа float32. |
embedding_table | Tensor типа float32. |
accumulator | Tensor типа float32. |
num_minibatches_per_physical_sparse_core | Tensor типа int32. |
table_name | string. |
clip_weight_min | Необязательный float. Значение по умолчанию: float('-inf'). |
clip_weight_max | Необязательный float. Значение по умолчанию: float('inf'). |
name | Имя операции (необязательно). |
| Возвращаемые значения |
Кортеж объектов Tensor (updated_embedding_table, updated_accumulator). |
updated_embedding_table | Tensor типа float32. |
updated_accumulator | Tensor типа float32. |