Browse Source

修改 paddle_to 中关于 device 类型的文档

tags/v1.0.0alpha
x54-729 3 years ago
parent
commit
b29b002bcc
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      fastNLP/core/utils/paddle_utils.py

+ 2
- 1
fastNLP/core/utils/paddle_utils.py View File

@@ -68,7 +68,8 @@ def paddle_to(data: "paddle.Tensor", device: Union[str, int, 'paddle.fluid.core_
该函数只是集成了 :func:`paddle.Tensor.cpu` 和 :func:`paddle.Tensor.cuda` 两个函数。

:param data: 要迁移的张量;
:param device: 目标设备,可以是 ``str`` 或 ``int`` 类型;
:param device: 目标设备,可以是 ``str`` 或 ``int`` 及 **paddle** 自己的 :class:`paddle.fluid.core_avx.Place`、
:class:`paddle.CPUPlace` 和 :class:`paddle.CUDAPlace` 类型;
:return: 迁移后的张量;
"""
if isinstance(device, paddle.fluid.core_avx.Place):


Loading…
Cancel
Save