Skip to content

toolkit.array.python

Functions:

as_scalar 

as_scalar(x: Any) -> float
Source code in src/toolkit/array/python/_cast.py 
 6
 7
 8
 9
10
11
12
13
def as_scalar(x: Any) -> float:
    if tk.is_jax(x):
        return x.item()
    if tk.is_numpy(x):
        return x.item()
    if tk.is_torch(x):
        return x.item()
    return x