Method __init__

diffusioninst/util/misc.py:288–290

(self, tensors, mask: Optional[Tensor])

class NestedTensor(object):
    def __init__(self, tensors, mask: Optional[Tensor]):
        self.tensors = tensors
        self.mask = mask

    def to(self, device):
        # type: (Device) -> NestedTensor # noqa