torsh_functional::manipulation::splitting

Enum SplitArg

pub enum SplitArg {
    Size(usize),
    Sections(usize),
    Indices(Vec<usize>),
}

Expand description

Split argument specification for tensor splitting operations

§Variants

§Size

Split into chunks of specified size along the dimension. The last chunk may be smaller if the dimension size is not evenly divisible.

§Sections

Split into the specified number of approximately equal sections. If not evenly divisible, earlier sections will be one element larger.

§Indices

Split at the specified indices, creating len(indices)+1 sub-tensors. Indices must be sorted and within bounds of the dimension.

Variants§

§

Size(usize)

Split into chunks of fixed size

§

Sections(usize)

Split into specified number of sections

§

Indices(Vec<usize>)

Split at specified indices

Trait Implementations§

Source §

impl Clone for SplitArg

Source §

fn clone(&self) -> SplitArg

Returns a duplicate of the value. Read more

1.0.0 · Source§

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source. Read more

Source §

impl Debug for SplitArg

Source §

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter. Read more

Auto Trait Implementations§

§

impl UnwindSafe for SplitArg

Blanket Implementations§

Source §

impl<T> Any for T
where T: 'static + ?Sized,

Source §

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more

Source §

impl<T> Borrow<T> for T
where T: ?Sized,

Source §

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more

Source §

impl<T> BorrowMut<T> for T
where T: ?Sized,

Source §

fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more

Source §

impl<T> CloneToUninit for T
where T: Clone,

Source §

unsafe fn clone_to_uninit(&self, dest: *mut u8)

🔬This is a nightly-only experimental API. (clone_to_uninit)

Performs copy-assignment from self to dest. Read more

Source §

impl<T> From<T> for T

Source §

fn from(t: T) -> T

Returns the argument unchanged.

Source §

impl<T, U> Into for T
where U: From<T>,

Source §

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

Source §

impl<T> IntoEither for T

Source §

fn into_either(self, into_left: bool) -> Either<Self, Self>

Converts self into a Left variant of Either<Self, Self> if into_left is true. Converts self into a Right variant of Either<Self, Self> otherwise. Read more

Source §

fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
where F: FnOnce(&Self) -> bool,

Converts self into a Left variant of Either<Self, Self> if into_left(&self) returns true. Converts self into a Right variant of Either<Self, Self> otherwise. Read more

Source §