Cupy создает массив с использованием других переменных ⇐ Python

[Anonymous]

Я пытаюсь создать сценарий обмена numpy/cupy, аналогичный этой внутренней реализации. Таким образом, используя что-то вроде бэкэнда импорта Util.Backend как bd, я могу создать bd.array(), который может переключаться между numpy и cupy. Но у меня было много проблем с этим, особенно с созданием массива.
После некоторых экспериментов выяснилось, что присваивание в виде:

Код: Выделить всё

import cupy as cp
A = cp.array([1, 2, 3])
B = cp.array([A[0], A[1], 3])

Приведет к ошибке:

Код: Выделить всё

TypeError: Implicit conversion to a NumPy array is not allowed. Please use `.get()` to construct a NumPy array explicitly.

Однако, если то же самое создание массива записано как:

Код: Выделить всё

B = cp.array([A[0], A[1], A[2]])

Тогда все становится совершенно нормально (что тоже странно, поскольку в приведенном выше примере я вообще не импортировал numpy, это почти похоже на создание [] сначала как массив numpy).
Аналогично,

Код: Выделить всё

s = 2
c = 3
one = 1.0
B = cp.array([s, c, one])

Все в порядке. Но если некоторые записи созданы по-разному, например:

Код: Выделить всё

s = cp.sin(2)
c = cp.cos(1)
one = 1.0
B = cp.array([s, c, one])

Тогда TypeError появится снова. Обратите внимание: если вместо cupy используется numpy, ни одно из описанных выше действий по созданию массива не приведет к ошибке.
Этот пост, похоже, указывает на то, что cupy не поддерживает данные смешанного типа. Но с помощью предложенного метода мне пришлось бы написать CParray.get(), чтобы выполнить преобразование. Это метод cupy, а не numpy, поэтому будут создаваться новые ошибки, если этот внутренний модуль работает numpy.
Можно ли найти способ создания массивов, чтобы cp и np были взаимозаменяемыми?

Подробнее здесь: https://stackoverflow.com/questions/793 ... -variables