prysm.mathops#
A submodule which allows the user to swap out the backend for mathematics.
- class prysm.mathops.BackendShim(src)#
Bases:
objectA shim that allows a backend to be swapped at runtime.
- prysm.mathops.set_backend_to_cupy()#
Convenience method to automatically configure prysm’s backend to cupy.
- prysm.mathops.set_backend_to_defaults()#
Convenience method to restore prysm’s default backend options.
- prysm.mathops.set_backend_to_pytorch()#
Convenience method to automatically configure prysm’s backend to PyTorch.
- prysm.mathops.jinc(r)#
Jinc.
The first zero of jinc occurs at r=pi
- Parameters:
r (number) – radial distance
- Returns:
the value of j1(x)/x for x != 0, 0.5 at 0
- Return type:
float
- prysm.mathops.is_odd(int)#
Determine if an interger is odd using binary operations.
- Parameters:
int (int) – an integer
- Returns:
true if odd, False if even
- Return type:
bool
- prysm.mathops.is_power_of_2(value)#
Check if a value is a power of 2 using binary operations.
- Parameters:
value (number) – value to check
- Returns:
true if the value is a power of two, False if the value is no
- Return type:
bool
Notes
c++ inspired implementation, see SO: https://stackoverflow.com/questions/29480680/finding-if-a-number-is-a-power-of-2-using-recursion
- prysm.mathops.sign(x)#
Sign of a number. Note only works for single values, not arrays.
- prysm.mathops.kronecker(i, j)#
Kronecker delta function, 1 if i = j, otherwise 0.
- prysm.mathops.gamma(n, m)#
Gamma function.