Exemplos de Ops.get_dropout_mask em Python

Exemplo n.º 1

def _get_drop_mask(ops: Ops, nO: int, rate: Optional[float]) -> Optional[Floats1d]:
    if rate is not None:
        mask = ops.get_dropout_mask((nO,), rate)
        return mask  # type: ignore
    return None

Exemplo n.º 2

Arquivo: staticvectors_v1.py Projeto: noemitth10/Learning-App

def _get_drop_mask(ops: Ops, nO: int,
                   rate: Optional[float]) -> Optional[Floats1d]:
    return ops.get_dropout_mask((nO, ), rate) if rate is not None else None