Нормализовать массив из 3 векторов, используя Numba и Cuda

Question

У меня есть большой массив трехмерных векторов, которые я хочу нормализовать, используя numba и cuda. Мой массив немного похож на этот

vectors = []
vectors.append([1.0, 1.0, 1.0])
vectors.append([-1.0, -1.0, -1.0])

Будучи новичком в numba и cuda, я безуспешно пробирался через блоги и учебные пособия, пытаясь написать функцию, которая может нормализовать мой массив векторов параллельно в gpu.

Может кто-нибудь показать мне, как это сделать или указать на полезную статью?

Answer 1

Вы можете использовать unit_vector ():

import transformations as trafo
import numpy as np

vectors.append([1.0, 1.0, 1.0])
vectors.append([-1.0, -1.0, -1.0])

normalized_vector= trafo.unit_vector(vectors, axis=1)