frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix torch.tensor warning in ONNX symbolic_opset10 export (#158835)

Browse Source 
Fix PyTorch tensor copying warning in ONNX export

## Problem

PyTorch ONNX exporter was generating a warning about incorrect tensor copying method:

```
UserWarning: To copy construct from a tensor, it is recommended to use sourceTensor.clone().detach() or sourceTensor.clone().detach().requires_grad_(True), rather than torch.tensor(sourceTensor).
```

Pull Request resolved: https://github.com/pytorch/pytorch/pull/158835
Approved by: https://github.com/justinchuby
This commit is contained in:
Alexander Novikov
2025-07-22 16:32:45 +00:00
committed by PyTorch MergeBot
parent 7d6f340238
commit 0971637c11
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
torch/onnx/symbolic_opset10.py
View File

@ -513,7 +513,9 @@ def _slice(
if is_none_value(list_or_value) and default_value is not None:
list_or_value = [default_value]
if isinstance(list_or_value, (list, torch.Tensor)):
if isinstance(list_or_value, torch.Tensor):
return g.op("Constant", value_t=list_or_value.clone().detach())
elif isinstance(list_or_value, list):
return g.op("Constant", value_t=torch.tensor(list_or_value))
rank = symbolic_helper._get_tensor_rank(list_or_value)