tensorflow如何使用多個CPU

Question

如何在TensorFlow（python API）中手動將多個設備分配給單個操作？

例如，爲了一個CPU分配給圖形，我會做：

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

如果我想分配CPU 0，1，2，3所有這同運？

看來我不能使用設備名稱列表作爲tf.device的arg。任何幫助或指導，將不勝感激。謝謝！

Answer 1

您可以multiple with contexts像這樣：

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