tf.raw_ops.XlaSparseDenseMatmulWithCsrInput

Псевдонимы

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

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

tf.compat.v1.raw_ops.XlaSparseDenseMatmulWithCsrInput

tf.raw_ops.XlaSparseDenseMatmulWithCsrInput(
    row_pointers,
    sorted_sample_ids,
    sorted_token_ids,
    sorted_gains,
    embedding_table,
    num_minibatches_per_physical_sparse_core,
    input_size,
    quantization_config_low,
    quantization_config_high,
    quantization_config_num_buckets,
    table_name,
    name=None
)

Аргументы
`row_pointers`	A `Tensor` типа `int32`.
`sorted_sample_ids`	A `Tensor` типа `int32`.
`sorted_token_ids`	A `Tensor` типа `int32`.
`sorted_gains`	A `Tensor` типа `float32`.
`embedding_table`	A `Tensor` типа `float32`.
`num_minibatches_per_physical_sparse_core`	A `Tensor` типа `int32`.
`input_size`	A `int`, который является `>= 0`.
`quantization_config_low`	A `float`.
`quantization_config_high`	A `float`.
`quantization_config_num_buckets`	A `int`, который является `>= 0`.
`table_name`	A `string`.
`name`	Имя операции (необязательно).

Возвращаемое значение
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/XlaSparseDenseMatmulWithCsrInput