nvflare.private.fed.utils.numproto module

NumPy ndarray to protobuf serialization and deserialization.

bytes_to_proto(data: bytes) → NDArray

Serializes a bytes into an NDArray protobuf message.

Parameters

data – bytes data

Returns

Returns an NDArray protobuf message.

ndarray_to_proto(nda: ndarray) → NDArray

Serializes a numpy array into an NDArray protobuf message.

Parameters

nda (np.ndarray) – numpy array to serialize.

Returns

Returns an NDArray protobuf message.

proto_to_bytes(nda_proto: NDArray) → bytes

Deserializes an NDArray protobuf message into bytes.

Parameters

nda_proto (NDArray) – bytes.

Returns

Returns bytes.

proto_to_ndarray(nda_proto: NDArray) → ndarray

Deserializes an NDArray protobuf message into a numpy array.

Parameters

nda_proto (NDArray) – NDArray protobuf message to deserialize.

Returns

Returns a numpy.ndarray.