Удаленный субъект генерирует опыт (cuda тензор хранится в графических процессорах). Могу ли я взять его и использовать без повторного копирования в ЦП?

Question

Я использую удаленных актеров Ray для генерации событий воспроизведения, а затем пробую некоторый опыт и возвращаю их в основную функцию и передаю их агенту RL для обучения.Могу ли я использовать их на графических процессорах и избежать копирования обратно на процессоры, а затем копировать на графические процессоры?