dpctl.tensor

dpctl.tensor provides a reference implementation of the Python Array API specification. The implementation uses data-parallel algorithms suitable for execution on accelerators, such as GPUs.

dpctl.tensor is written using C++ and SYCL and oneAPI extensions implemented in Intel(R) oneAPI DPC++ compiler.

This module contains:

• Array object usm_ndarray

• array creation functions

• array manipulation functions

• elementwise functions

• indexing functions

• introspection functions

• linear algebra functions

• searching functions

• set functions

• sorting functions

• statistical functions

• utility functions