未验证 提交 67a4de68 编写于 作者: L Leo Chen 提交者: GitHub

Add return when input is tensor (#33570)

* add return when input is tensor

* fix typo
上级 294dfd23
......@@ -134,8 +134,9 @@ def to_tensor(data, dtype=None, place=None, stop_gradient=True):
)
elif isinstance(data, paddle.Tensor):
data = data._copy_to(place, False)
ata = _handle_dtype(data, dtype)
data = _handle_dtype(data, dtype)
data.stop_gradient = stop_gradient
return data
elif isinstance(data, (core.LoDTensor, core.Tensor)):
# Note(zhouwei25): should't expose it to users, just for internal use.
# convert core.Tensor/core.LoDTensor to VarBase first
......@@ -145,6 +146,7 @@ def to_tensor(data, dtype=None, place=None, stop_gradient=True):
data = data._copy_to(place, False)
data = _handle_dtype(data, dtype)
data.stop_gradient = stop_gradient
return data
else:
raise TypeError(
"Can't constructs a 'paddle.Tensor' with data type {}, data type must be scalar|list|tuple|numpy.ndarray|paddle.Tensor".
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册