__init__.py 98 Bytes
Newer Older
bailuo's avatar
readme  
bailuo committed
1
2
3
from .masked_conv import MaskedConv2d, masked_conv2d

__all__ = ['masked_conv2d', 'MaskedConv2d']