Struct RawTensor 

pub struct RawTensor {
    pub data: Vec<u8>,
    pub dtype: u8,
    pub shape: Vec<usize>,
}

Represents a tensor in Cake protocol.

Fields

data: Vec<u8>

Tensor data.

dtype: u8

The data type as a compact u8 tag (see dtype_to_u8 / u8_to_dtype).

shape: Vec<usize>

The tensor shape.

Implementations

impl RawTensor

pub fn from_tensor(x: &Tensor) -> Self

Convert x into a RawTensor.

pub fn to_tensor(&self, device: &Device) -> Result<Tensor>

Convert the raw tensor in a Tensor allocated on the given device.

Trait Implementations

impl Debug for RawTensor

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

Formats the value using the given formatter.

impl<'a_, C_: Context> Readable<'a_, C_> for RawTensor

fn read_from<R_: Reader<'a_, C_>>(_reader_: &mut R_) -> Result<Self, C_::Error>

fn minimum_bytes_needed() -> usize

fn read_from_buffer(buffer: &'a [u8]) -> Result<Self, <C as Context>::Error>

where Self: DefaultContext<Context = C>, C: Default,

Deserializes from a given buffer.

fn read_with_length_from_buffer( buffer: &'a [u8], ) -> (Result<Self, <C as Context>::Error>, usize)

where Self: DefaultContext<Context = C>, C: Default,

Deserializes from a given buffer while also returning the amount of bytes consumed.

fn read_from_buffer_copying_data( buffer: &[u8], ) -> Result<Self, <C as Context>::Error>

where Self: DefaultContext<Context = C>, C: Default,

Deserializes from a given buffer.

fn read_with_length_from_buffer_copying_data( buffer: &[u8], ) -> (Result<Self, <C as Context>::Error>, usize)

where Self: DefaultContext<Context = C>, C: Default,

Deserializes from a given buffer while also returning the amount of bytes consumed.

fn read_from_stream_unbuffered( stream: impl Read, ) -> Result<Self, <C as Context>::Error>

where Self: DefaultContext<Context = C>, C: Default,

Reads from a given stream without any buffering.

fn read_from_stream_buffered( stream: impl Read, ) -> Result<Self, <C as Context>::Error>

where Self: DefaultContext<Context = C>, C: Default,

Reads from a given stream with internal buffering.

fn read_from_file(path: impl AsRef<Path>) -> Result<Self, <C as Context>::Error>

where Self: DefaultContext<Context = C>, C: Default,

fn read_from_buffer_with_ctx( context: C, buffer: &'a [u8], ) -> Result<Self, <C as Context>::Error>

fn read_with_length_from_buffer_with_ctx( context: C, buffer: &'a [u8], ) -> (Result<Self, <C as Context>::Error>, usize)

fn read_from_buffer_copying_data_with_ctx( context: C, buffer: &[u8], ) -> Result<Self, <C as Context>::Error>

fn read_with_length_from_buffer_copying_data_with_ctx( context: C, buffer: &[u8], ) -> (Result<Self, <C as Context>::Error>, usize)

fn read_with_length_from_buffer_copying_data_with_ctx_mut( context: &mut C, buffer: &[u8], ) -> (Result<Self, <C as Context>::Error>, usize)

fn read_from_stream_unbuffered_with_ctx<S>( context: C, stream: S, ) -> Result<Self, <C as Context>::Error>

where S: Read,

fn read_from_stream_buffered_with_ctx<S>( context: C, stream: S, ) -> Result<Self, <C as Context>::Error>

where S: Read,

fn read_from_file_with_ctx( context: C, path: impl AsRef<Path>, ) -> Result<Self, <C as Context>::Error>

impl<C_: Context> Writable<C_> for RawTensor

fn write_to<T_: ?Sized + Writer<C_>>( &self, _writer_: &mut T_, ) -> Result<(), C_::Error>

fn write_to_buffer( &self, buffer: &mut [u8], ) -> Result<(), <C as Context>::Error>

where Self: DefaultContext<Context = C>, C: Default,

fn write_to_vec(&self) -> Result<Vec<u8>, <C as Context>::Error>

where Self: DefaultContext<Context = C>, C: Default,

fn write_to_stream<S>(&self, stream: S) -> Result<(), <C as Context>::Error>

where S: Write, Self: DefaultContext<Context = C>, C: Default,

fn write_to_file( &self, path: impl AsRef<Path>, ) -> Result<(), <C as Context>::Error>

where Self: DefaultContext<Context = C>, C: Default,

fn write_to_buffer_with_ctx( &self, context: C, buffer: &mut [u8], ) -> Result<(), <C as Context>::Error>

fn write_to_buffer_with_ctx_mut( &self, context: &mut C, buffer: &mut [u8], ) -> Result<(), <C as Context>::Error>

fn write_to_vec_with_ctx( &self, context: C, ) -> Result<Vec<u8>, <C as Context>::Error>

fn write_to_vec_with_ctx_mut( &self, context: &mut C, ) -> Result<Vec<u8>, <C as Context>::Error>

fn write_to_stream_with_ctx<S>( &self, context: C, stream: S, ) -> Result<(), <C as Context>::Error>

where S: Write,

fn write_to_file_with_ctx( &self, context: C, path: impl AsRef<Path>, ) -> Result<(), <C as Context>::Error>

fn bytes_needed(&self) -> Result<usize, <C as Context>::Error>