tf.fixed_size_partitioner(num_shards, axis=0)See the guide: Variables > Variable Partitioners for Sharding
Partitioner to specify a fixed number of shards along given axis.
num_shards: int, number of shards to partition variable.axis: int, axis to partition on.A partition function usable as the partitioner argument to variable_scope, get_variable, and get_partitioned_variable_list.
Defined in tensorflow/python/ops/partitioned_variables.py.
© 2017 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/api_docs/python/tf/fixed_size_partitioner