warp.optim.linear.aslinearoperator

warp.optim.linear.aslinearoperator(A, batch_offsets=None)

Cast the dense or sparse matrix A as a LinearOperator.

A must be of one of the following types:

• warp.sparse.BsrMatrix

• two-dimensional warp.array; then A is assumed to be a dense matrix

• one-dimensional warp.array; then A is assumed to be a diagonal matrix

• warp.optim.linear.LinearOperator; no casting necessary, batch_offsets is ignored

Parameters:

• A (array | BsrMatrix | LinearOperator) – The matrix to wrap.

• batch_offsets (array | None) – Optional array of shape (B+1,) partitioning scalar degrees of freedom into B independent subproblems (see LinearOperator).

Return type:

LinearOperator