nvflare.app_opt.pt.decomposers module

class TensorDecomposer[source]

Bases: Decomposer

decompose(target: Tensor, manager: DatumManager | None = None) → Any[source]

Decompose the target into types supported by msgpack or classes with decomposers registered.

Msgpack supports primitives, bytes, memoryview, lists, dicts.

Parameters:

  • target – The instance to be serialized

  • manager – Datum manager to store externalized datum

Returns:

The decomposed serializable objects

recompose(data: Any, manager: DatumManager | None = None) → Tensor[source]

Reconstruct the object from decomposed components.

Parameters:

  • data – The decomposed component

  • manager – Datum manager to internalize datum

Returns:

The reconstructed object

supported_type()[source]

Returns the type/class supported by this decomposer.

Returns:

The class (not instance) of supported type