Trait dfdx::tensor::SplitTape

pub trait SplitTape {
    type Tape;
    type NoTape: Clone + PutTape<Self::Tape, Output = Self>;

    // Required method
    fn split_tape(self) -> (Self::NoTape, Self::Tape);
}

Remove the tape from a tensor

Required Associated Types

type Tape

The type of tape the tensor has now

type NoTape: Clone + PutTape<Self::Tape, Output = Self>

Required Methods

fn split_tape(self) -> (Self::NoTape, Self::Tape)

Splits tape off of self

let a: Tensor<Rank1<5>, f32, _, OwnedTape<f32, _>> = dev.zeros().traced(grads);
let (a, tape): (Tensor<_, _, _, NoneTape>, OwnedTape<f32, _>) = a.split_tape();

Implementors

impl<S: Shape, E: Clone, D: Storage<E>, T> SplitTape for Tensor<S, E, D, T>

type Tape = T

type NoTape = Tensor<S, E, D, NoneTape>