tf.distribute.experimental.partitioners.FixedShardsPartitioner

Разделитель, который распределяет фиксированное количество фрагментов.

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

tf.distribute.experimental.partitioners.FixedShardsPartitioner(
    num_shards
)

Примеры:

# standalone usage:
partitioner = FixedShardsPartitioner(num_shards=2)
partitions = partitioner(tf.TensorShape([10, 3]), tf.float32)
[2, 1]

# use in ParameterServerStrategy
# strategy = tf.distribute.experimental.ParameterServerStrategy(
#   cluster_resolver=cluster_resolver, variable_partitioner=partitioner)

Аргументы
`num_shards`	`int`, количество фрагментов для разбиения.

Методы

`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`	a `tf.TensorShape`, форма для разбиения.
`dtype`	a `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/FixedShardsPartitioner