Source code for pymoose.computation.values

from dataclasses import dataclass

import numpy as np



[docs]@dataclass
class Value:
    pass




[docs]@dataclass
class Constant(Value):
    pass




[docs]@dataclass
class ShapeConstant(Constant):
    value: tuple




[docs]@dataclass
class StringConstant(Constant):
    value: str




[docs]@dataclass
class BytesConstant(Constant):
    value: bytes




[docs]@dataclass
class TensorConstant(Constant):
    value: np.ndarray

    def __hash__(self):
        return hash(self.value.tobytes())

    def __eq__(self, other):
        return isinstance(other, TensorConstant) and np.all(self.value == other.value)




[docs]@dataclass
class IntConstant(Constant):
    value: int




[docs]@dataclass
class FloatConstant(Constant):
    value: float