Spec-Zone.ru › PyTorch 1

torch.linalg.matrix_ранг

torch.linalg.matrix_rank(A, *, atol=None, rtol=None, hermitian=False, out=None) → Tensor

Вычисляет числовой ранг матрицы.

Ранг матрицы вычисляется как количество сингулярных значений (или собственных значений по модулю, когда hermitian= True) превышающих порог max⁡(atol,σ1∗rtol)\max(\text{atol}, \sigma_1 * \text{rtol}), где σ1\sigma_1 — наибольшее сингулярное значение (или собственное значение).

Поддерживает входные типы float, double, cfloat и cdouble. Также поддерживает пакеты матриц, и если A — пакет матриц, то выходные данные имеют те же размерности пакета.

Если hermitian= True, A предполагается эрмитовой, если комплексной, или симметричной, если вещественной, но это не проверяется внутри. Вместо этого используется только нижняя треугольная часть матрицы в вычислениях.

Если rtol не указан и A — матрица размерностей (m, n), относительная погрешность устанавливается в rtol=max⁡(m,n)ε\text{rtol} = \max(m, n) \varepsilon и ε\varepsilon — значение эпсилон для типа A (см. finfo). Если rtol не указан, а atol указан больше нуля, то rtol устанавливается в ноль.

Если atol или rtol — torch.Tensor, её размерность должна быть совместима с размерностью сингулярных значений A, возвращаемых torch.linalg.svdvals().

Примечание

Эта функция имеет аналог NumPy linalg.matrix_rank(A, tol, hermitian=False). Однако использование позиционного аргумента tol устарело в пользу atol и rtol.

Примечание

Ранг матрицы вычисляется с помощью разложения по сингулярным значениям torch.linalg.svdvals() если hermitian= False (по умолчанию) и с помощью разложения по собственным значениям torch.linalg.eigvalsh() при hermitian= True. Когда входные данные находятся на устройстве CUDA, эта функция синхронизирует это устройство с процессором.

Параметры:
  • A (Тензор) – тензор с формой (*, m, n) где * — нуль или более размерностей пакета.
  • tol (float, Тензор, необязательно) – [NumPy Compat] Псевдоним для atol. По умолчанию: None.
Ключевые аргументы:
  • atol (float, Тензор, необязательно) – абсолютное значение погрешности. При None считается равным нулю. По умолчанию: None.
  • rtol (float, Тензор, необязательно) – относительное значение погрешности. См. выше для значения, когда None. По умолчанию: None.
  • hermitian (bool) – указывает, является ли A эрмитовой, если комплексной, или симметричной, если вещественной. По умолчанию: False.
  • out (Тензор, необязательно) – выходной тензор. Игнорируется, если None. По умолчанию: None.

Примеры:

>>> A = torch.eye(10)
>>> torch.linalg.matrix_rank(A)
tensor(10)
>>> B = torch.eye(10)
>>> B[0, 0] = 0
>>> torch.linalg.matrix_rank(B)
tensor(9)

>>> A = torch.randn(4, 3, 2)
>>> torch.linalg.matrix_rank(A)
tensor([2, 2, 2, 2])

>>> A = torch.randn(2, 4, 2, 3)
>>> torch.linalg.matrix_rank(A)
tensor([[2, 2, 2, 2],
        [2, 2, 2, 2]])

>>> A = torch.randn(2, 4, 3, 3, dtype=torch.complex64)
>>> torch.linalg.matrix_rank(A)
tensor([[3, 3, 3, 3],
        [3, 3, 3, 3]])
>>> torch.linalg.matrix_rank(A, hermitian=True)
tensor([[3, 3, 3, 3],
        [3, 3, 3, 3]])
>>> torch.linalg.matrix_rank(A, atol=1.0, rtol=0.0)
tensor([[3, 2, 2, 2],
        [1, 2, 1, 2]])
>>> torch.linalg.matrix_rank(A, atol=1.0, rtol=0.0, hermitian=True)
tensor([[2, 2, 2, 1],
        [1, 2, 2, 2]])

© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/1.13/generated/torch.linalg.matrix_rank.html

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API