with文で複数のウィンドウを並べるには

Question

with文で複数のウィンドウを並べる方法はないでしょうか？
gpuをwith文でいくつにも分割しようとしましたが、

num_gpus=8
for i in range(num_gpus):
with tf.device('/gpu:%d' % i)

AttributeError: 'DeviceSpec' object has no attribute 'split'

そこで以下のように訂正しましたが、

gpus =[tf.device('/gpu:%d' % i) for i in range(NUMN_GPUS)]
    for gpu in gpus:
        with gpu:

AttributeError: __enter__

with文で

with tf.device('/gpu:0'), tf.device('/gpu:1')~ tf.device('/gpu:8')

と複数並べる方法はないでしょうか？
ご教授お願いします

Answer 1

contextlib.ExitStackが良いと思います。以下の例では、withを抜ける時にtf.device('/gpu:8')～tf.device('/gpu:0')の__exit__が呼ばれます。

Comments

[trafalbad]

回答ありがとうございます。再度質問失礼します。
gpuのmodel を呼ぶとき以下のようにすればよろしいのでしょうか？

with ExitStack() as stack:
for dev in map('/gpu:{}'.format, range(9)):
stack.enter_context(tf.device(dev))
parallel_model= multi_gpu_model(train_model, gpus=NUMN_GPUS)

[YouheiSakurai]

tfは触らないので分からないです。回答はあくまで複数のコンテキストをwithする方法です。