burn::backend::ndarray

Macro execute_with_float_dtype

macro_rules! execute_with_float_dtype {
    (($lhs:expr, $rhs:expr), $op:expr) => { ... };
    (($lhs:expr, $rhs:expr), $element:ident, $op:expr) => { ... };
    (($lhs:expr, $rhs:expr) => $op:expr) => { ... };
    ($tensor:expr, $op:expr) => { ... };
    ($tensor:expr, $element:ident, $op:expr) => { ... };
    ($tensor:expr => $op:expr) => { ... };
    ($tensor:expr, $element:ident => $op:expr) => { ... };
}
Expand description

Macro to execute an operation a given element type.

§Panics

Since there is no automatic type cast at this time, binary operations for different floating point precision data types will panic with a data type mismatch.