NamedFormats#

class nvmath.sparse.ust.NamedFormats[source]#

A number of pre-defined common tensor formats (direct or using builders). Clients can always construct these or other tensor formats directly.

Attributes

BatchedCSR = [batch, i, j] -> (batch: <LevelFormat.DENSE>, i: <LevelFormat.DENSE>, j: <LevelFormat.COMPRESSED>)#

BatchedDIAINonUniform = [batch, i, j] -> (batch: <LevelFormat.DENSE>, (j - i): <LevelFormat.COMPRESSED>, i: <LevelFormat.RANGE>)#

BatchedDIAIUniform = [batch, i, j] -> ((j - i): <LevelFormat.COMPRESSED>, batch: <LevelFormat.DENSE>, i: <LevelFormat.RANGE>)#

CCOL = [i, j] -> (j: <LevelFormat.COMPRESSED>, i: <LevelFormat.DENSE>)#

COO = [i, j] -> (i: (<LevelFormat.COMPRESSED>, <LevelProperty.NONUNIQUE>), j: <LevelFormat.SINGLETON>)#

CROW = [i, j] -> (i: <LevelFormat.COMPRESSED>, j: <LevelFormat.DENSE>)#

CSC = [i, j] -> (j: <LevelFormat.DENSE>, i: <LevelFormat.COMPRESSED>)#

CSR = [i, j] -> (i: <LevelFormat.DENSE>, j: <LevelFormat.COMPRESSED>)#

DCSC = [i, j] -> (j: <LevelFormat.COMPRESSED>, i: <LevelFormat.COMPRESSED>)#

DCSR = [i, j] -> (i: <LevelFormat.COMPRESSED>, j: <LevelFormat.COMPRESSED>)#

DIAI = [i, j] -> ((j - i): <LevelFormat.COMPRESSED>, i: <LevelFormat.RANGE>)#

DIAJ = [i, j] -> ((j - i): <LevelFormat.COMPRESSED>, j: <LevelFormat.RANGE>)#

DenseMatrixLeft = [i, j] -> (j: <LevelFormat.DENSE>, i: <LevelFormat.DENSE>)#

DenseMatrixRight = [i, j] -> (i: <LevelFormat.DENSE>, j: <LevelFormat.DENSE>)#

DenseVector = [i] -> (i: <LevelFormat.DENSE>)#

Scalar = [] -> ()#

SkewDIAI = [i, j] -> ((i + j): <LevelFormat.COMPRESSED>, i: <LevelFormat.RANGE>)#

SkewDIAJ = [i, j] -> ((i + j): <LevelFormat.COMPRESSED>, j: <LevelFormat.RANGE>)#

SparseVector = [i] -> (i: <LevelFormat.COMPRESSED>)#

b = batch#

i = i#

j = j#

k = k#

Methods

static BSCLeft(blocksize)[source]#

BSCLeft matrix format.

Parameters:: blocksize – a pair specifying the blocksize in the i’th and j’th dimensions.

static BSCRight(blocksize)[source]#

BSCRight matrix format.

Parameters:: blocksize – a pair specifying the blocksize in the i’th and j’th dimensions.

static BSCRightd(blocksize, batch_dim=0, dense_dim=0)[source]#

static BSR3(blocksize)[source]#

3D BSR format.

Parameters:: blocksize – a triple specifying the blocksize in the i’th, j’th, and k’th dimensions.

static BSRLeft(blocksize)[source]#

BSRLeft matrix format.

Parameters:: blocksize – a pair specifying the blocksize in the i’th and j’th dimensions.

static BSRRight(blocksize)[source]#

BSRRight matrix format.

Parameters:: blocksize – a pair specifying the blocksize in the i’th and j’th dimensions.

static BSRRightd(blocksize, batch_dim=0, dense_dim=0)[source]#

static BlockVector(blocksize)[source]#

Block vector format.

Parameters:: blocksize – an integer specifying the blocksize in the i’th dimension.

static COOd(sparse_dim, dense_dim=0)[source]#

static CSCd(batch_dim=0, dense_dim=0)[source]#

static CSFd(dim)[source]#

static CSRd(batch_dim=0, dense_dim=0)[source]#

static DELTA(delta)[source]#

Delta compression matrix format.

Parameters:: delta – number of bits for delta

static DensedLeft(dim)[source]#

static DensedRight(dim)[source]#