mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
Previous implementation of the `numpy()` method returns `fp64` when the tensor is `fp32`. This is unexpected but seems to be caused by calling `__array__(dtype=None)` on the numpy array. I updated the implementation to implement the `numpy()` method explicitly and added tests to guard the behavior. This needs to be cherry-picked into torch 2.5 Pull Request resolved: https://github.com/pytorch/pytorch/pull/136162 Approved by: https://github.com/gramalingam, https://github.com/xadupre |
||
|---|---|---|
| .. | ||
| diagnostics | ||
| exporter | ||
| fx | ||
| __init__.py | ||
| _exporter_legacy.py | ||
| _lazy_import.py | ||
| io_adapter.py | ||
| jit_utils.py | ||
| onnx_proto_utils.py | ||
| onnxruntime.py | ||
| registration.py | ||