tf.distribute.experimental.partitioners.Partitioner

Базовый класс для разделителей: все разделители наследуются от этого класса.

Разделители должны реализовывать метод __call__ со следующей сигнатурой:

def __call__(self, shape, dtype, axis=0):
  # Partitions the given `shape` and returns the partition results.
  # See docstring of `__call__` method for the format of partition results.

Методы

`call`

Посмотреть исходный код

__call__(
    shape, dtype, axis=0
)

Разбивает заданный shape и возвращает результаты разбиения.

Пример разделителя, выделяющего фиксированное число шардов:

partitioner = FixedShardsPartitioner(num_shards=2)
partitions = partitioner(tf.TensorShape([10, 3], tf.float32), axis=0)
print(partitions) # [2, 0]

Аргументы
`shape`	объект `tf.TensorShape`, форма для разбиения.
`dtype`	объект `tf.dtypes.Dtype`, указывающий тип значения разбиения.
`axis`	Ось, вдоль которой выполняется разбиение. По умолчанию: самая внешняя ось.

Возвращает
Список целых чисел, представляющих количество разделов по каждой оси, где i-е значение соответствует i-й оси.

© 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/distribute/experimental/partitioners/Partitioner