Función torch.linalg.matrix_rank de PyTorch

Pytorch torch 参考手册Manual de referencia de Pytorch torch

torch.linalg.matrix_rankEs una función del módulo de álgebra lineal de PyTorch que se utiliza para calcular el rango de una matriz. El rango de una matriz representa el número de filas o columnas linealmente independientes de la matriz.

Definición de la función

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

Parámetros:

  • A(Tensor): La matriz de entrada.
  • tol(float, opcional): Umbral de valor singular.
  • hermitian(bool, opcional): Si es True, la matriz se considera Hermitiana (simétrica). Por defecto es False.
  • out(Tensor, opcional): Tensor de salida.

Valor de retorno:

  • torch.Tensor: Devuelve el rango de la matriz.

Ejemplo de uso

Ejemplo

import torch

# Crear una matriz de rango completo
A = torch.tensor([[1.0, 2.0, 3.0],
                  [4.0, 5.0, 6.0],
                  [7.0, 8.0, 9.0]])

# Calcular el rango de la matriz
rank = torch.linalg.matrix_rank(A)

print("Matriz A:")
print(A)
print("Rango de la matriz:", rank)

El resultado es:

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

Ejemplo - Matriz de rango completo

import torch

# Crear una matriz de rango completo
B = torch.tensor([[1.0, 2.0],
                  [3.0, 4.0]])

rank_B = torch.linalg.matrix_rank(B)
print("Matriz B:")
print(B)
print("Rango de la matriz:", rank_B)

Pytorch torch 参考手册Manual de referencia de Pytorch torch

Otras extensiones