CuPy

Communauté d’assistance GitHub Outils d’optimisation

CuPy est une implémentation d’un tableau multidimensionnel compatible NumPy sur CUDA. CuPy se compose de cupy.ndarray, la classe principale de tableau multidimensionnel, et de nombreuses fonctions qui y sont associées. Il prend en charge un sous-ensemble de l’interface numpy.ndarray.

Caractéristiques

Indexation de base (indexation par ints, slices, newaxes et Ellipsis)

La plupart des indexations avancées (sauf certains modèles d’indexation avec masques booléens)

Types de données (dtypes) : bool_, int8, int16, int32, int64, uint8, uint16, uint32, uint64, float16, float32, float64, complexe64, complexe128

La plupart des routines de création de tableaux (empty, ones_like, diag, etc.)

La plupart des routines de manipulation de tableaux (remodelage, rollaxis, concaténation, etc.)

Tous les opérateurs avec diffusion

Toutes les fonctions universelles pour les opérations élément par élément (sauf celles pour les nombres complexes)

Fonctions d’algèbre linéaire, y compris produit (dot, matmul, etc.) et décomposition (cholesky, svd, etc.), accélérées par cuBLAS et cuSOLVER

Transformée de Fourier rapide multidimensionnelle (FFT), accélérée par cuFFT

Réduction le long des axes (somme, max, argmax, etc.)

Site officiel

Tutoriel et documentation

Enter your contact information to continue reading