jcreinhold

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
grid attention blocks for gated attention networks
Based on: https://github.com/ozan-oktay/Attention-Gated-Networks
Author: Jacob Reinhold (jacob.reinhold@jhu.edu)
"""

__all__ = ['GridAttentionBlock2d',
           'GridAttentionBlock3d']