def _check_shape(s1, s2):
    """check s1 == s2 if both are not None"""
    if s1 and s2 and s1 != s2:
        raise ValueError('Shape mismatch detected. ' + str(s1) + ' v.s. ' +
            str(s2))
