Функция PyTorch torch.linalg.matrix_rank

Pytorch torch 参考手册Справочное руководство PyTorch torch

torch.linalg.matrix_rankЭто функция в модуле линейной алгебры PyTorch, используемая для вычисления ранга матрицы. Ранг матрицы представляет собой число линейно независимых строк или столбцов матрицы.

Определение функции

torch.linalg.matrix_rank(A, tol=None, hermitian=False, out=None)

Parameters:

  • A(Tensor): входная матрица.
  • tol(float, необязательный): порог сингулярного значения.
  • hermitian(bool, необязательный): если True, предполагается, что матрица эрмитова (симметричная). По умолчанию False.
  • out(Tensor, необязательный): Выходной тензор.

Возвращаемое значение:

  • torch.Tensor: Возвращает ранг матрицы.

Пример использования

Пример

import torch

# Создать матрицу полного ранга
A = torch.tensor([[1.0, 2.0, 3.0],
                  [4.0, 5.0, 6.0],
                  [7.0, 8.0, 9.0]])

# Вычислить ранг матрицы
rank = torch.linalg.matrix_rank(A)

print("Матрица A:")
print(A)
print("Ранг матрицы:", rank)

Вывод:

矩阵 A:
tensor([[1., 2., 3.],
        [4., 5., 6.],
        [7., 8., 9.]])
矩阵的秩: tensor(2)

Пример — матрица полного ранга

import torch

# Создать матрицу полного ранга
B = torch.tensor([[1.0, 2.0],
                  [3.0, 4.0]])

rank_B = torch.linalg.matrix_rank(B)
print("Матрица B:")
print(B)
print("Ранг матрицы:", rank_B)

Pytorch torch 参考手册Справочное руководство PyTorch torch

Другие расширения