tf.compat.v1.fixed_size_partitioner

Partitioner to specify a fixed number of shards along given axis.

Args
num_shards int, number of shards to partition variable.
axis int, axis to partition on.
Returns
A partition function usable as the partitioner argument to variable_scope and get_variable.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/compat/v1/fixed_size_partitioner