Trait rai_core::ops::ArangeArgs

pub trait ArangeArgs<D: Type>: Debug {
    // Required method
    fn stop(&self) -> D::Repr;

    // Provided methods
    fn start(&self) -> D::Repr { ... }
    fn step(&self) -> D::Repr { ... }
    fn size(&self) -> usize { ... }
}

Represents the arguments for the arange function.

Required Methods

fn stop(&self) -> D::Repr

Returns the stop value for the arange function.

Provided Methods

fn start(&self) -> D::Repr

Returns the start value for the arange function.

fn step(&self) -> D::Repr

Returns the step value for the arange function.

fn size(&self) -> usize

Returns the size of the resulting Tensor for the arange function.

Implementations on Foreign Types

impl ArangeArgs<F16> for (f16, f16)

fn start(&self) -> f16

fn stop(&self) -> f16

impl ArangeArgs<F16> for (f16, f16, f16)

fn start(&self) -> f16

fn stop(&self) -> f16

fn step(&self) -> f16

impl ArangeArgs<F16> for f16

fn stop(&self) -> f16

impl ArangeArgs<F32> for (f32, f32)

fn start(&self) -> f32

fn stop(&self) -> f32

impl ArangeArgs<F32> for (f32, f32, f32)

fn start(&self) -> f32

fn stop(&self) -> f32

fn step(&self) -> f32

impl ArangeArgs<F32> for f32

fn stop(&self) -> f32

impl ArangeArgs<F64> for (f64, f64)

fn start(&self) -> f64

fn stop(&self) -> f64

impl ArangeArgs<F64> for (f64, f64, f64)

fn start(&self) -> f64

fn stop(&self) -> f64

fn step(&self) -> f64

impl ArangeArgs<F64> for f64

fn stop(&self) -> f64

impl ArangeArgs<U8> for (u8, u8)

fn start(&self) -> u8

fn stop(&self) -> u8

impl ArangeArgs<U8> for (u8, u8, u8)

fn start(&self) -> u8

fn stop(&self) -> u8

fn step(&self) -> u8

impl ArangeArgs<U8> for u8

fn stop(&self) -> u8

impl ArangeArgs<U32> for (u32, u32)

fn start(&self) -> u32

fn stop(&self) -> u32

impl ArangeArgs<U32> for (u32, u32, u32)

fn start(&self) -> u32

fn stop(&self) -> u32

fn step(&self) -> u32

impl ArangeArgs<U32> for u32

fn stop(&self) -> u32

Implementors