This library is intended as a pure Rust
Tensor library which runs on the GPU everywhere.
In order to achieve it, it build upon the
wgpu crate which can target any platform supporting
DX12. It can run even on a Raspberry Pi 4 GPU.
The API entry point is the
A GpuTensorView share the same data as the original Tensor, but can have different shapes and strides For example, the original shape could be [2, 2] and the GpuTensorView could be [1, 2, 2]
Same as GpuTensorView but mutable
A Tensor is an N dimensional data structure. This is the entry point for most of the API
of this crate. This is normally backed by GPU memory and its device chosen using the current
default of the
A view into the original