docs/api/api_python/hal/mindspore.hal.memory_allocated.rst - master - envoy.istio/mindspore

docs/api/api_python/hal/mindspore.hal.memory_allocated.rst

16 строк590 B

shen_haochen

doc formatting

11 июн 2024, 12:02

11 июн 2024, 12:02a780bf6

mindspore.hal.memory_allocated
==============================

.. py:function:: mindspore.hal.memory_allocated(device_target=None)

    返回当前真实被Tensor占用的内存大小。

    .. note::
        - 若用户不指定 `device_target` ，将此参数设置为当前已经设置的后端类型。
        - 对于 `CPU` 后端，固定返回0。

    参数：
        - **device_target** (str，可选) - 用户指定的后端类型，必须是 ``"CPU"`` ， ``"GPU"`` 以及 ``"Ascend"`` 的其中一个。默认值：``None``。

    返回：
        int，单位为Byte。