sparseTensor(sparseTensorWithDescriptor:tensors:shape:name:)
Creates a sparse tensor representation.
Declaration
func sparseTensor(sparseTensorWithDescriptor sparseDescriptor: MPSGraphCreateSparseOpDescriptor, tensors inputTensorArray: [MPSGraphTensor], shape: [NSNumber], name: String?) -> MPSGraphTensorParameters
- sparseDescriptor:
A sparseDescriptor.
- inputTensorArray:
An array of input tensors as [sparseVals, indexTensor0, indexTensor1].
- shape:
The shape of the sparse tensor.
- name:
A name for the operation.
Return Value
A valid MPSGraphTensor object
Discussion
sparseVals corresponds to non zero values in matrix. indexTensor0 and indexTensor1 are indices used for indexing into sparse data structure. For COO, indexTensor0 is x index and indexTensor1 is y index . For CSC, indexTensor0 and indexTensor1 correspond to rowIndex and colStarts respectively. For CSR, indexTensor0 and indexTensor1 correspond to colIndex and rowStarts respectively. You must set input tensors appropriately for each sparse storage type.