tf.raw_ops.GetMinibatchesInCsrWithPhysicalReplica

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

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

Подробнее см. в руководстве по миграции.

tf.compat.v1.raw_ops.GetMinibatchesInCsrWithPhysicalReplica

tf.raw_ops.GetMinibatchesInCsrWithPhysicalReplica(
    program_key,
    row_ids,
    col_ids,
    gains,
    splits,
    id_counts,
    sample_count,
    num_replica,
    max_minibatches_per_sc,
    max_ids_per_chip_per_sample,
    table_vocab_size,
    feature_width,
    num_sc_per_chip,
    table_name,
    mini_batch_in_csr,
    name=None
)

Аргументы
`program_key`	`Tensor` типа `string`.
`row_ids`	`Tensor` типа `int32`.
`col_ids`	`Tensor` типа `int32`.
`gains`	`Tensor` типа `float32`.
`splits`	`Tensor` типа `int64`.
`id_counts`	`Tensor` типа `int32`.
`sample_count`	`int`, который является `>= 1`.
`num_replica`	`int`, который является `>= 1`.
`max_minibatches_per_sc`	`int`, который является `>= 1`.
`max_ids_per_chip_per_sample`	`int`, который является `>= 1`.
`table_vocab_size`	`int`, который является `>= 1`.
`feature_width`	`int`, который является `>= 1`.
`num_sc_per_chip`	`int`, который является `>= 1`.
`table_name`	`string`.
`mini_batch_in_csr`	`string`.
`name`	Имя операции (необязательно).

Возвращаемые значения
Кортеж из `Tensor` объектов (row_pointers, sorted_sample_ids, sorted_token_ids, sorted_gains, row_pointers_unpadded_size, ids_unpadded_size, num_minibatches_per_physical_sparse_core).
`row_pointers`	`Tensor` типа `int32`.
`sorted_sample_ids`	`Tensor` типа `int32`.
`sorted_token_ids`	`Tensor` типа `int32`.
`sorted_gains`	`Tensor` типа `float32`.
`row_pointers_unpadded_size`	`Tensor` типа `int32`.
`ids_unpadded_size`	`Tensor` типа `int32`.
`num_minibatches_per_physical_sparse_core`	`Tensor` типа `int32`.

© 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/GetMinibatchesInCsrWithPhysicalReplica