cupy.who

cupy.who(vardict=None)

打印给定字典中的 CuPy 数组。

打印出 vardict 中存在的所有 ndarray 的名称、形状、字节数和类型。

如果未传入字典或 vardict 为 None，则返回 globals() 字典中的 CuPy 数组（命名空间中的所有 CuPy 数组）。

参数:

vardict – (None 或 dict) 可能包含 ndarray 的字典。如果指定 None，则默认为 globals()

示例

>>> a = cupy.arange(10)
>>> b = cupy.ones(20)
>>> cupy.who()
Name            Shape            Bytes            Type
===========================================================

a               10               80               int64
b               20               160              float64

Upper bound on total bytes  =       240
>>> d = {'x': cupy.arange(2.0),
... 'y': cupy.arange(3.0), 'txt': 'Some str',
... 'idx':5}
>>> cupy.who(d)
Name            Shape            Bytes            Type
===========================================================

x               2                16               float64
y               3                24               float64

Upper bound on total bytes  =       40