关于什么时候切换numpy或者cuda.tensor的问题

[pangkun248]

RT，我看到你的代码中时不时的要把数据从numpy上切换到cuda.tensor上或者从cuda.tensor切换到numpy上面，这一点我在chenyuntc/simple-faster-rcnn-pytorch的代码中也有发现。请问你知道有关这方面的技巧吗？让代码运行更快的技巧。就是应该在什么时候从numpy->cuda，而什么时候从cuda->numpy。或者如果你知道相关论文博客链接的话也可以。