Python tensorflow如何使用多个cpu

如何在TensorFlow（python API）中手动将多个设备分配给一个操作

例如，为了将一个cpu分配给图形，我将执行以下操作：

with tf.device("/cpu:0"):
    # Define operation here

如果我想将cpu 0、1、2、3全部分配给同一个op怎么办

似乎我不能使用设备名称列表作为tf.device的参数。任何帮助或指导都将不胜感激。谢谢

您可以这样做：

---

默认情况下，自动使用所有cpu，cpu:0表示计算机上的所有cpu

with tf.device("/cpu:0") as cpu0, tf.device("/cpu:1") as cpu1:
    pass