mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
68b9daa9bf
Summary: Adds `torch.linalg.norm` function that matches the behavior of `numpy.linalg.norm`. Additional changes: * Add support for dimension wrapping in `frobenius_norm` and `nuclear_norm` * Fix `out` argument behavior for `nuclear_norm` * Fix issue where `frobenius_norm` allowed duplicates in `dim` argument * Add `_norm_matrix` Closes https://github.com/pytorch/pytorch/issues/24802 Pull Request resolved: https://github.com/pytorch/pytorch/pull/42749 Reviewed By: ngimel Differential Revision: D23336234 Pulled By: mruberry fbshipit-source-id: f0aba3089a3a0bf856aa9c4215e673ff34228fac
17 lines
235 B
ReStructuredText
17 lines
235 B
ReStructuredText
.. role:: hidden
|
|
:class: hidden-section
|
|
|
|
torch.linalg
|
|
============
|
|
|
|
Common linear algebra operations.
|
|
|
|
.. automodule:: torch.linalg
|
|
.. currentmodule:: torch.linalg
|
|
|
|
Functions
|
|
---------
|
|
|
|
.. autofunction:: det
|
|
.. autofunction:: norm
|