Struct OrtOwnedTensor

pub struct OrtOwnedTensor<'t, 'm, T, D>
where
    T: TypeToTensorElementDataType + Debug + Clone,
    D: Dimension,
    'm: 't,
{ /* private fields */ }

Tensor containing data owned by the ONNX Runtime C library, used to return values from inference.

This tensor type is returned by the Session::run() method. It is not meant to be created directly.

The tensor hosts an ndarray::ArrayView of the data on the C side. This allows manipulation on the Rust side using ndarray without copying the data.

OrtOwnedTensor implements the std::deref::Deref trait for ergonomic access to the underlying ndarray::ArrayView.

Implementations

impl<'t, 'm, T, D> OrtOwnedTensor<'t, 'm, T, D>

where T: TypeToTensorElementDataType + Debug + Clone, D: Dimension,

pub fn softmax(&self, axis: Axis) -> Array<T, D>

where D: RemoveAxis, T: NdFloat + SubAssign + DivAssign,

Apply a softmax on the specified axis

Trait Implementations

impl<'t, 'm, T, D> Debug for OrtOwnedTensor<'t, 'm, T, D>

where T: TypeToTensorElementDataType + Debug + Clone + Debug, D: Dimension + Debug, 'm: 't,

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

Formats the value using the given formatter.

impl<'t, 'm, T, D> Drop for OrtOwnedTensor<'t, 'm, T, D>

where T: TypeToTensorElementDataType + Debug + Clone, D: Dimension, 'm: 't,

fn drop(&mut self)

Executes the destructor for this type.

impl<'t, 'm, T, D> Deref for OrtOwnedTensor<'t, 'm, T, D>

where T: TypeToTensorElementDataType + Debug + Clone, D: Dimension,

type Target = ArrayBase<ViewRepr<&'t T>, D>

The resulting type after dereferencing.

fn deref(&self) -> &Self::Target

Dereferences the value.