tf.train.experimental.ShardByTaskPolicy

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

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

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

Атрибуты
`description`

Методы

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

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

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

Возвращает
Список словарей шардов, содержащих тензоры. [ {checkpoint key: {slice_spec: tensor} } ]

© 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