Sparse tensors in Gorgonia

test_sparse.go

package main

import (
	"fmt"

	. "gorgonia.org/tensor"
)

func main() {
	xs := []int{1, 2, 6, 8}
	ys := []int{1, 2, 1, 6}
	vals := []float32{3, 1, 4, 1}

	S := CSCFromCoord(Shape{9, 7}, xs, ys, vals)
	T2 := New(WithShape(9, 7), Of(Float32)) // dense

	Result, _ := Add(S, T2)
	fmt.Printf("When adding a sparse tensor to a dense tensor, the result is of %T:\n=============================================================================\n%+#s\n", Result, Result)
	Result, _ = Add(T2, S)
	fmt.Printf("And vice versa - %T\n=========================\n%+#s\n", Result, Result)

	// Output:
	// When adding a sparse tensor to a dense tensor, the result is of *tensor.Dense:
	// =============================================================================
	// Matrix (9, 7) [7 1]
	// ⎡0  0  0  0  0  0  0⎤
	// ⎢0  3  0  0  0  0  0⎥
	// ⎢0  0  1  0  0  0  0⎥
	// ⎢0  0  0  0  0  0  0⎥
	// ⎢0  0  0  0  0  0  0⎥
	// ⎢0  0  0  0  0  0  0⎥
	// ⎢0  4  0  0  0  0  0⎥
	// ⎢0  0  0  0  0  0  0⎥
	// ⎣0  0  0  0  0  0  1⎦

	// And vice versa - *tensor.Dense
	// =========================
	// Matrix (9, 7) [7 1]
	// ⎡0  0  0  0  0  0  0⎤
	// ⎢0  3  0  0  0  0  0⎥
	// ⎢0  0  1  0  0  0  0⎥
	// ⎢0  0  0  0  0  0  0⎥
	// ⎢0  0  0  0  0  0  0⎥
	// ⎢0  0  0  0  0  0  0⎥
	// ⎢0  4  0  0  0  0  0⎥
	// ⎢0  0  0  0  0  0  0⎥
	// ⎣0  0  0  0  0  0  1⎦

}