0703344409780e55a055c03ff590c44d35cc10d0,thinc/layers/dropout.py,,,#,9

Before Change

// TODO: How to type the "sub-functions"?
// TODO: improve this and make array types more specific
InTArray = TypeVar("InTArray", bound=Array)
InputLengths = TypeVar("InputLengths", bound=Array)
InTList = List[InTArray]
InTRagged = Tuple[InTArray, InputLengths]
InT = Union[InTArray, InTList, InTRagged]
OutTArray = TypeVar("OutTArray", bound=Array)
OutputLengths = TypeVar("OutputLengths", bound=Array)
OutTList = List[OutTArray]
OutTRagged = Tuple[OutTArray, OutputLengths]
OutT = Union[OutTArray, OutTList, OutTRagged]


def Dropout(rate: float = 0.0) -> Model[InT, OutT]:

After Change

// TODO: improve this and make array types more specific?
InT = Union[Array, List[Array], Ragged]
OutT = Union[Array, List[Array], Ragged]

Instances