def eye_mask(self, shape):
"""
Build a mask using np.eye.
"""
return ~np.eye(*shape, dtype=bool)
微信
公众号
扫码关注公众号
微信
公众号