Struct GradientMessage 

pub struct GradientMessage {
    pub id: String,
    pub data: Vec<u8>,
    pub shape: Vec<usize>,
    pub dtype: String,
    pub checksum: u64,
    pub metadata: HashMap<String, String>,
}

Gradient message for federated learning

Fields

id: String

Gradient identifier (e.g., layer name or tensor CID)

data: Vec<u8>

Gradient data (compressed)

shape: Vec<usize>

Shape of the gradient tensor

dtype: String

Data type (f32, f16, etc.)

checksum: u64

Checksum for verification

metadata: HashMap<String, String>

Metadata (e.g., learning rate, batch size)

Implementations

impl GradientMessage

pub fn new( id: impl Into<String>, data: Vec<u8>, shape: Vec<usize>, dtype: impl Into<String>, ) -> Self

Create a new gradient message

pub fn verify(&self) -> bool

Verify checksum

pub fn with_metadata( self, key: impl Into<String>, value: impl Into<String>, ) -> Self

Add metadata

pub fn num_elements(&self) -> usize

Get total elements in gradient

pub fn size_bytes(&self) -> usize

Estimate size in bytes

Trait Implementations

impl Clone for GradientMessage

fn clone(&self) -> GradientMessage

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for GradientMessage

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for GradientMessage

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for GradientMessage

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.