'Returns the mask by which to multiply the parts of the embedding layer.
In this version, we have no weights to apply.'

def _get_mask(self, idx, in_data):

    mask = (idx >= 0)
    mask = np.expand_dims(mask, 2)
    mask = np.repeat(mask, self._proj_dim, axis=2)
    return mask
