Class Flows

flowmap/flow/flow_predictor.py:17–21

@dataclass
class Flows(Manipulable):
    forward: Float[Tensor, "batch pair flow_height flow_width 2"]
    backward: Float[Tensor, "batch pair flow_height flow_width 2"]
    forward_mask: Float[Tensor, "batch pair flow_height flow_width"]
    backward_mask: Float[Tensor, "batch pair flow_height flow_width"]


T = TypeVar("T")