19 lines
314 B
Python
19 lines
314 B
Python
import numpy as np
|
|
|
|
_f = np.float32
|
|
|
|
def _check(a):
|
|
assert isinstance(a, np.ndarray) or type(a) == _f, type(a)
|
|
assert a.dtype == _f, a.dtype
|
|
return a
|
|
|
|
_0 = _f(0)
|
|
_1 = _f(1)
|
|
_2 = _f(2)
|
|
_inv2 = _f(1/2)
|
|
_sqrt2 = _f(np.sqrt(2))
|
|
_invsqrt2 = _f(1/np.sqrt(2))
|
|
_pi = _f(np.pi)
|
|
|
|
__all__ = [o for o in locals()]
|