Struct TensorBuilder 

pub struct TensorBuilder { /* private fields */ }

Implementations

impl TensorBuilder

pub fn with_shape(self, shape: impl AsRef<[i64]>) -> Self

pub fn with_typ(self, typ: ONNXTensorElementDataType) -> Self

pub fn with_memory_info(self, mem_info: MemoryInfo) -> Self

pub fn borrow(self, input: &impl TensorTrait) -> Result<Value<'_>>

pub fn alloc(self, alloc: &impl AllocatorTrait) -> Result<Value<'_>>

Trait Implementations

impl Default for TensorBuilder

fn default() -> Self

Return TensorBuilder { shape: Default::default(), typ: ONNXTensorElementDataType::ONNX_TENSOR_ELEMENT_DATA_TYPE_UNDEFINED, mem_info: Default::default() }