Crate burn_ndarray 

Burn ndarray backend.

Macros

cat_with_dtype

Macro to execute a cat operation on a given set of element types.

execute_with_dtype

Macro to execute an operation on a given element type.

execute_with_float_dtype

Macro to execute an operation a given element type. Only handles float types.

execute_with_int_dtype

Macro to execute an operation a given element type. Only handles int types.

execute_with_numeric_dtype

Macro to execute an operation a given element type. Only handles numeric types

iter_parmulti-threads

Macro for iterating in parallel.

iter_range_parmulti-threads

Macro for iterating over a range in parallel.

iter_slice_parmulti-threads

Macro for iterating in parallel.

reshape

Reshapes an array into a tensor.

run_parmulti-threads

Macro for running a function in parallel.

to_typed_dims

Converts a slice of usize to a typed dimension.

Structs

NdArray

Tensor backend that uses the ndarray crate for executing tensor operations.

NdArrayQTensor

A quantized tensor for the ndarray backend.

Enums

NdArrayDevice

The device type for the ndarray backend.

NdArrayTensor

Tensor primitive used by the ndarray backend.

Traits

AddAssignElement

The addition assignment operator implemented for ndarray elements.

ExpElement

A element for ndarray backend that supports exp ops.

FloatNdArrayElement

A float element for ndarray backend.

IntNdArrayElement

An int element for ndarray backend.

NdArrayElement

A general element for ndarray backend.

QuantElement

A quantized element for the ndarray backend.

Type Aliases

SharedArray

Concrete storage type for ndarray (owned with COW semantics via Arc)