Struct StreamTensor

pub struct StreamTensor(/* private fields */);

A stream tensor is used in streaming module. It can either contain an actual tensor or be empty.

Implementations

impl StreamTensor

pub fn empty() -> Self

pub fn from_tensor(tensor: Tensor) -> Self

pub fn shape(&self) -> Option<&Shape>

pub fn cat2<D: Dim>(&self, rhs: &Self, dim: D) -> Result<Self>

pub fn seq_len<D: Dim>(&self, dim: D) -> Result<usize>

pub fn reset(&mut self)

pub fn narrow<D: Dim>( &self, dim: D, offset: usize, len: usize, ) -> Result<StreamTensor>

pub fn split<D: Dim>(&self, dim: D, lhs_len: usize) -> Result<(Self, Self)>

Splits the Streaming Tensor on the time axis dim with the first lhs_len elements returned in the first output and the remaining in the second output.

pub fn as_option(&self) -> Option<&Tensor>

pub fn apply<M: Module>(&self, m: &M) -> Result<Self>

Trait Implementations

impl Clone for StreamTensor

fn clone(&self) -> StreamTensor

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for StreamTensor

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

Formats the value using the given formatter.

impl From<()> for StreamTensor

fn from(_value: ()) -> Self

Converts to this type from the input type.

impl From<Option<Tensor>> for StreamTensor

fn from(value: Option<Tensor>) -> Self

Converts to this type from the input type.

impl From<Tensor> for StreamTensor

fn from(value: Tensor) -> Self

Converts to this type from the input type.