| 123456789101112131415161718192021 |
- # 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, mask, requires_grad=False):
- return MaskedTensor(data, mask, requires_grad)
- def as_masked_tensor(data, mask):
- return MaskedTensor._from_values(data, mask)
|
