aten

[kindlychung]

repo: https://github.com/zdevito/ATen

ATen: A TENsor library for C++11

ATen is fundamentally a tensor library, on top of which almost all other Python and C++ interfaces in PyTorch are built. It provides a core Tensor class, on which many hundreds of operations are defined. Most of these operations have both CPU and GPU implementations, to which the Tensor class will dynamically dispatch based on its type.

Useful for deep learning projects.