You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
25 lines
605 B
25 lines
605 B
# 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)
|