tf.raw_ops.XlaSparseDenseMatmulGradWithSgdAndCsrInput
tf.raw_ops.XlaSparseDenseMatmulGradWithSgdAndCsrInput(
row_pointers,
sorted_sample_ids,
sorted_token_ids,
sorted_gains,
activation_gradients,
learning_rate,
embedding_table,
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. |
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 типа float32. |