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])