Crate burn_ndarray

Burn ndarray backend.

Macros

execute_with_float_dtype

Macro to execute an operation a given element type.

new_tensor_float

Macro to create a new float tensor based on the element type.

reshape

Reshapes an array into a tensor.

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.

NdArrayTensor

Tensor primitive used by the ndarray backend.

Enums

NdArrayDevice

The device type for the ndarray backend.

NdArrayTensorFloat

Float tensor primitive.

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.