Crate burn_ndarray

Source
Expand description

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.