def check_unary_nan(self, name, xp, dtype):
a = xp.array(
[-3, numpy.NAN, -1, numpy.NAN, 0, numpy.NAN, numpy.inf],
dtype=dtype)
return getattr(xp, name)(a)
微信
公众号
扫码关注公众号