This looks like a great project! I have a question: Why is it necessary to specify the device for every Tensor? Wouldn't it be possible to set the device once

This looks like a great project! I have a question: Why is it necessary to specify the device for every Tensor? Wouldn't it be possible to set the device once and then all allocations are made to that device?