当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python tf.compat.v1.distribute.experimental.ParameterServerStrategy.experimental_make_numpy_dataset用法及代码示例

用法

experimental_make_numpy_dataset(
    numpy_input, session=None
)

参数

  • numpy_input 将转换为数据集的 NumPy 输入数组的嵌套。请注意,Numpy 数组的列表是堆叠的,因为这是正常的 tf.data.Dataset 行为。
  • session (仅限 TensorFlow v1.x 图执行)用于初始化的会话。

返回

为通过 numpy 数组提供的输入创建一个 tf.data.Dataset。

这避免了将numpy_input 添加为图形中的大常数,并将数据复制到将处理输入的机器或机器上。

请注意,您可能需要将 tf.distribute.Strategy.experimental_distribute_dataset 与返回的数据集一起使用,以进一步将其与策略一起分发。

例子:

numpy_input = np.ones([10], dtype=np.float32)
dataset = strategy.experimental_make_numpy_dataset(numpy_input)
dist_dataset = strategy.experimental_distribute_dataset(dataset)

相关用法


注:本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.compat.v1.distribute.experimental.ParameterServerStrategy.experimental_make_numpy_dataset。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。