cupyx.scipy.linalg.tri

cupyx.scipy.linalg.tri(N, M=None, k=0, dtype=None)

Construct (N, M) matrix filled with ones at and below the k-th diagonal. The matrix has A[i,j] == 1 for i <= j + k.

Parameters
  • N (int) – The size of the first dimension of the matrix.

  • M (int, optional) – The size of the second dimension of the matrix. If M is None, M = N is assumed.

  • k (int, optional) – Number of subdiagonal below which matrix is filled with ones. k = 0 is the main diagonal, k < 0 subdiagonal and k > 0 superdiagonal.

  • dtype (dtype, optional) – Data type of the matrix.

Returns

Tri matrix.

Return type

cupy.ndarray