[−][src]Crate tensor_compute
A WebGPU based Tensor Library
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
either Vulkan
, Metal
or DX12
. It can run even on a Raspberry Pi 4 GPU.
The API entry point is the Tensor
structure.
Modules
prelude | |
traits | |
traits | |
utils |
Macros
s |
Structs
CpuTensor | |
GpuStore | |
GpuTensor | |
GpuTensorView | 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] |
GpuTensorViewMut | Same as GpuTensorView but mutable |
LinearIndexer | |
ShapeStrides | |
SliceRangeInfo | |
Tensor | 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 |
TensorView | A view into the original |
Traits
AsShaderInput | |
Constructable | |
CpuTransferable | |
GpuAllocated | |
MutShapeStrideTrait | |
ShapeStrideTrait |