Struct EagerSliceBuilder 

pub struct EagerSliceBuilder<'a> { /* private fields */ }

Rank-preserving eager tensor slicing builder.

Unspecified axes are kept whole. Range selections become one Slice operation; host-known position selections become Gather/index_select operations.

Examples

use tenferro_ad::{EagerRuntime, EagerTensor, Tensor};

let ctx = EagerRuntime::new();
let x = EagerTensor::from_tensor_in(
    Tensor::from_vec_col_major(vec![3, 4], vec![0.0_f64; 12]).unwrap(),
    ctx,
).unwrap();
let y = x.slice_builder().axis(0, 0..2).axis_step(1, 0..4, 2).apply().unwrap();
assert_eq!(y.shape(), &[2, 2]);

Implementations

impl<'a> EagerSliceBuilder<'a>

pub fn axis(self, axis: usize, range: Range<usize>) -> Self

Add an exclusive-end range selection for one axis.

Examples

use tenferro_ad::{EagerRuntime, EagerTensor, Tensor};

let ctx = EagerRuntime::new();
let x = EagerTensor::from_tensor_in(
    Tensor::from_vec_col_major(vec![4], vec![1.0_f64, 2.0, 3.0, 4.0]).unwrap(),
    ctx,
).unwrap();
let y = x.slice_builder().axis(0, 1..3).apply().unwrap();
assert_eq!(y.shape(), &[2]);

pub fn axis_step(self, axis: usize, range: Range<usize>, step: usize) -> Self

Add an exclusive-end strided range selection for one axis.

Examples

use tenferro_ad::{EagerRuntime, EagerTensor, Tensor};

let ctx = EagerRuntime::new();
let x = EagerTensor::from_tensor_in(
    Tensor::from_vec_col_major(vec![5], vec![1.0_f64, 2.0, 3.0, 4.0, 5.0]).unwrap(),
    ctx,
).unwrap();
let y = x.slice_builder().axis_step(0, 0..5, 2).apply().unwrap();
assert_eq!(y.shape(), &[3]);

pub fn take_axis(self, axis: usize, indices: &[usize]) -> Self

Add a host-known position selection for one axis.

Examples

use tenferro_ad::{EagerRuntime, EagerTensor, Tensor};

let ctx = EagerRuntime::new();
let x = EagerTensor::from_tensor_in(
    Tensor::from_vec_col_major(vec![3], vec![1.0_f64, 2.0, 3.0]).unwrap(),
    ctx,
).unwrap();
let y = x.slice_builder().take_axis(0, &[2, 0]).apply().unwrap();
assert_eq!(y.shape(), &[2]);

pub fn apply(self) -> Result<EagerTensor>

Build and apply the requested slice/take operations.

Examples

use tenferro_ad::{EagerRuntime, EagerTensor, Tensor};

let ctx = EagerRuntime::new();
let x = EagerTensor::from_tensor_in(
    Tensor::from_vec_col_major(vec![4], vec![1.0_f64, 2.0, 3.0, 4.0]).unwrap(),
    ctx,
).unwrap();
let y = x.slice_builder().axis(0, 1..4).apply().unwrap();
assert_eq!(y.shape(), &[3]);

Trait Implementations

impl<'a> Clone for EagerSliceBuilder<'a>

fn clone(&self) -> EagerSliceBuilder<'a>

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<'a> Debug for EagerSliceBuilder<'a>

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

Formats the value using the given formatter.