gorgonia.org/tensor@v0.9.24/example_dense_reduction_test.go (about)

     1  package tensor
     2  
     3  import "fmt"
     4  
     5  func Example_sum_Sliced() {
     6  	T := New(WithShape(4, 4), WithBacking([]int{
     7  		1, 2, 3, 4,
     8  		5, 6, 7, 8,
     9  		1, 2, 3, 4,
    10  		5, 6, 7, 8,
    11  	}))
    12  	s, _ := T.Slice(S(1, 3), S(1, 3))
    13  	sum, _ := Sum(s)
    14  
    15  	fmt.Printf("T:\n%v\nsliced:\n%v\nSum: %v", T, s, sum)
    16  
    17  	// Output:
    18  	// T:
    19  	// ⎡1  2  3  4⎤
    20  	// ⎢5  6  7  8⎥
    21  	// ⎢1  2  3  4⎥
    22  	// ⎣5  6  7  8⎦
    23  	//
    24  	// sliced:
    25  	// ⎡6  7⎤
    26  	// ⎣2  3⎦
    27  	//
    28  	// Sum: 18
    29  
    30  }