Использование gpu_from_host в scan()

#python #gpu #theano

#python #графический процессор #theano

Вопрос:

Я хочу использовать scan() для создания циклов, которые выполняются на графическом процессоре. Однако, если я запускаю функцию, результат получается из type numpy.ndarray , что указывает на то, что результат передается на хост.

Как я могу убедиться, что данные не передаются на хост-машину? Я не могу найти пример, где gpu_from_host используется для этой цели.

Возможно ли это сделать?