Cómo obtener el rango de una array en PyTorch

En este artículo, vamos a discutir cómo obtener el rango de una array en PyTorch. podemos obtener el rango de una array utilizando el método torch.linalg.matrix_rank().

método torch.linalg.matrix_rank()

El método matrix_rank() acepta una array y un lote de arrays como entrada. Este método devuelve un nuevo tensor con el rango de las arrays de entrada y si la entrada es un lote de arrays, el tensor de salida también tiene las mismas dimensiones de lote que la entrada. la siguiente sintaxis se usa para obtener el rango de una array en PyTorch.

Sintaxis: torch.linalg.matrix_rank(mat)

Parámetros:

• mat: Esta es nuestra Array de entrada o un lote de arrays.

Devoluciones: devuelve un tensor con el rango de la array de entrada.

Ejemplo 1:

El siguiente programa es para entender cómo obtener el rango de array.

# import torch module
import torch
  
# define a tensor
tens = torch.tensor([[0.5322, 0.9232, 0.232],
                     [0.1112, 0.2323, 0.2611],
                     [0.7556, 0.1217, 0.5435]])
  
# display tensor/matrix
print("\n Input Matrix: ")
print(tens)
  
# get the rank of input matrix
r = torch.linalg.matrix_rank(tens)
  
# display rank of matrix
print("\n Rank of input matrix: ", r)

Producción:

 

Ejemplo 2:

El siguiente programa es para saber como obtener el rango de un lote de arrays.

# import torch module
import torch
  
# define a batch of matrix
# the below code create 2 matrix of 4x3
Batch = torch.rand(2, 4, 3)
  
# Display matrices
print("\n Batch of Matrices: ")
print(Batch)
  
# get ranks of the matrices
r = torch.linalg.matrix_rank(Batch)
  
# Display Result
print("\n Batch of matrices rank: ")
print(r)

Producción: