多GPU的简洁实现

http://zh-v2.d2l.ai/chapter_computational-performance/multiple-gpus-concise.html

有个疑问,为什么net是放在了0和1两个GPU上,但当train的时候,却只把X和y移到了GPU[0]上呢?谢谢!