tf.train.experimental.ShardByTaskPolicy

Политика разделения тензоров на фрагменты на основе задачи спецификации устройства.

Наследуется от: ShardingCallback

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

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

Атрибуты
`description`

Методы

__call__(
    shardable_tensors: Sequence[tf.train.experimental.ShardableTensor]
) -> Sequence[sharding_util.TensorSliceDict]

Обработчик для разделения тензоров на фрагменты на основе задачи спецификации устройства.

Аргументы
`shardable_tensors`	Список ShardableTensors.

Возвращаемые значения
Список словарей фрагментов, содержащих тензоры. [ {ключ контрольной точки: {спецификация среза: тензор} } ]

© 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/train/experimental/ShardByTaskPolicy