Module bshare
Source - ToBurn
- Trait for converting ndarray to burn tensor (generic over Float/Int and
dimensionality)
- ToNalgebraFloat
- Trait for converting
burn
tensor to nalgebra::DMatrix
or
nalgebra::DVector
(Float type) - ToNalgebraInt
- Trait for converting
burn
tensor to nalgebra::DMatrix
or
nalgebra::DVector
(Int type) - ToNdArray
- Trait for converting burn tensor to ndarray (generic over Float/Int and
dimensionality)
- tensor_to_data_float
- Convert a burn float tensor to a Vec
- tensor_to_data_int
- Convert a burn int tensor to a Vec