pytorch
340531f2 - [ONNX] Do not use `numpy` in ONNX opsets (#65188)

Commit
3 years ago
[ONNX] Do not use `numpy` in ONNX opsets (#65188) Summary: Replace `torch.tensor([numpy.arange(a, b, c)])` with `torch.arange(a, b, c).unsqueeze(0)` Replace `tuple(numpy.add(a, b))` with `tuple( x + y for (x, y) in zip(a, b)` As `numpy` is an optional dependency, it shouldn't be used in PyTorch core by default Pull Request resolved: https://github.com/pytorch/pytorch/pull/65188 Reviewed By: mruberry Differential Revision: D31009490 Pulled By: malfet fbshipit-source-id: 528e48f055bf9ac1de1fd7e94c0be41915df9a0b
Author
Parents
Loading