__init__.py 98 Bytes
Newer Older
1
from .masked_conv import masked_conv2d, MaskedConv2d
2
3

__all__ = ['masked_conv2d', 'MaskedConv2d']