Python tf.compat.v1.fixed_size_partitioner用法及代码示例

分区器指定沿给定轴的固定数量的分片。

用法

tf.compat.v1.fixed_size_partitioner(
    num_shards, axis=0
)

迁移到 TF2

警告：这个 API 是为 TensorFlow v1 设计的。继续阅读有关如何从该 API 迁移到本机 TensorFlow v2 等效项的详细信息。见TensorFlow v1 到 TensorFlow v2 迁移指南有关如何迁移其余代码的说明。

此 API 在 TF2 中已弃用。在 TF2 中，分区器不再是通过 tf.Variable 声明的变量的一部分。 ParameterServer Training 处理变量的分区。 fixed_size_partitioner 对应的 TF2 分区器类是 tf.distribute.experimental.partitioners.FixedShardsPartitioner 。

查看迁移指南，了解 TF1 和 TF2 在处理变量和损失方面的差异。

前：

x = tf.compat.v1.get_variable(
    "x", shape=(2,), partitioner=tf.compat.v1.fixed_size_partitioner(2)
  )

后：

partitioner = (
      tf.distribute.experimental.partitioners.FixedShardsPartitioner(
          num_shards=2)
  )
  strategy = tf.distribute.experimental.ParameterServerStrategy(
                 cluster_resolver=cluster_resolver,
                 variable_partitioner=partitioner)

  with strategy.scope():
    x = tf.Variable([1.0, 2.0])