Sparse tensor#

Mars tensor supports sparse tensor, unfortunately, only 2-D sparse tensors are available for now. Multi-dimensional sparse tensors will be supported later.

Functions to create sparse tensor#

mars.tensor.tensor

mars.tensor.zeros

Return a new tensor of given shape and type, filled with zeros.

mars.tensor.eye

Return a 2-D tensor with ones on the diagonal and zeros elsewhere.

mars.tensor.identity

Return the identity tensor.

mars.tensor.random.randint

Return random integers from low (inclusive) to high (exclusive).