Как выполнить сплайн-интерполяцию cubi c на python с CuPy?

Question

Я пишу код с использованием графического процессора, чтобы продолжать выполнять кубическую c сплайн-интерполяцию много раз. Я знаю, как это сделать на numpy, например, используя

scipy.interpolate.splrep

или

scipy.interpolate.interp1d(kind='cubic')

. Interp1d - это то, что я сейчас использую для массивов numpy. Но мне нужно запустить их на CuPy.

Но как мне это сделать на CuPy? У меня есть х-значения и у-значения. И у меня также есть массив с новыми значениями х. Код, который я сейчас пишу, будет вычислять новые значения y для новых значений x.

Answer 1

Я бы начал с исходного кода scipy.interpolate.CubicSpline, который является чистым python, и заменил бы np.something на cupy.something один за другим