Skip to content

tensorshare.Backend

Constants for tensorshare.serialization.

Backend

Bases: str, Enum

Supported backends.

Source code in src/tensorshare/serialization/constants.py 
 7
 8
 9
10
11
12
13
14
class Backend(str, Enum):
    """Supported backends."""

    FLAX = "flax"
    NUMPY = "numpy"
    PADDLEPADDLE = "paddlepaddle"
    # TENSORFLOW = "tensorflow"
    TORCH = "torch"

TensorType

Bases: str, Enum

Tensor types.

Source code in src/tensorshare/serialization/constants.py 
17
18
19
20
21
22
23
24
class TensorType(str, Enum):
    """Tensor types."""

    FLAX = "jaxlib.xla_extension.ArrayImpl"
    NUMPY = "numpy.ndarray"
    PADDLEPADDLE = "paddle.Tensor"
    # TENSORFLOW = "tensorflow.Tensor"
    TORCH = "torch.Tensor"
Last update: 2023-08-20
Created: 2023-08-20