tf.device() 也可以传递函数名而不是设备字符串。在这种情况下,该函数必须返回设备字符串。此函数允许使用复杂的算法将变量放在不同的设备上。例如,TensorFlow 在 tf.train.replica_device_setter() 中提供循环设备设置器,我们将在下一节中讨论。
tf.device()
tf.train.replica_device_setter()