Crate burn_ndarray
source ·Expand description
Burn ndarray backend.
Macros
- Macro for iterating in parallel.
- Macro for iterating over a range in parallel.
- Reshapes an array into a tensor.
- Macro for running a function in parallel.
- Converts a slice of usize to a typed dimension.
Structs
- Tensor backend that uses the ndarray crate for executing tensor operations.
- Tensor primitive used by the ndarray backend.
Enums
- The device type for the ndarray backend.
Traits
- A float element for ndarray backend.