Crate burn_ndarray 

Burn ndarray backend.

Macros

cat_with_dtype

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

execute_with_dtype

Macro to execute an operation 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

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