matrixBandPart function

VARP matrixBandPart(
  1. VARP input,
  2. VARP lower,
  3. VARP upper
)

Copies a variable setting everything outside a central band in each innermost matrix.

Arguments:

  • input: Rank k variable.
  • num_lower: Number of subdiagonals to keep. If negative, keep entire lower triangle.
  • num_upper: Number of superdiagonals to keep. If negative, keep entire upper triangle.

Returns:

  • Output: Rank k variable of the same shape as input. The extracted banded tensor.

Implementation

VARP matrixBandPart(VARP input, VARP lower, VARP upper) =>
    VARP.fromPointer(C.mnn_expr_MatrixBandPart(input.ptr, lower.ptr, upper.ptr));