| 123456789101112131415161718192021222324 |
- # Copyright (c) Meta Platforms, Inc. and affiliates
- from .core import MaskedTensor
- __all__ = [
- "as_masked_tensor",
- "masked_tensor",
- ]
- # These two factory functions are intended to mirror
- # torch.tensor - guaranteed to be a leaf node
- # torch.as_tensor - differentiable constructor that preserves the autograd history
- def masked_tensor(
- data: object, mask: object, requires_grad: bool = False
- ) -> MaskedTensor:
- return MaskedTensor(data, mask, requires_grad)
- def as_masked_tensor(data: object, mask: object) -> MaskedTensor:
- return MaskedTensor._from_values(data, mask)
|
