mul_sparse_sparse#
- iskra.sparse.mul_sparse_sparse(a: SparseTensor, b: SparseTensor) SparseTensor[SOURCE]#
- iskra.sparse.mul_sparse_sparse(a: Tensor, b: SparseTensor) SparseTensor
- iskra.sparse.mul_sparse_sparse(a: SparseTensor, b: Tensor) SparseTensor
- iskra.sparse.mul_sparse_sparse(a: Tensor, b: Tensor) Tensor
Elementwise product of two sparse COO tensors.
- Parameters:
a (
SparseTensor|Tensor) – First tensor to multiply.b (
SparseTensor|Tensor) – Second tensor to multiply.
- Raises:
ValueError – Two tensors must have the same shape.
- Returns:
(
SparseTensor|Tensor) – Elementwise product of the two tensors.