Trait burn_tensor::MovedimArgs
source · pub trait MovedimArgs {
// Required method
fn into_dim_vec<const D: usize>(self) -> Vec<usize>;
}Expand description
Trait used for movedim arguments
Required Methods§
sourcefn into_dim_vec<const D: usize>(self) -> Vec<usize>
fn into_dim_vec<const D: usize>(self) -> Vec<usize>
Converts into a set of dimensions Vec<usize> for the tensor.movedim() function
Object Safety§
This trait is not object safe.