In [1]:

    

import (
    "fmt"
    "math"

    "gonum.org/v1/gonum/mat"
)


    

In [2]:

    

// Create two matrices of the same size, a and b.
a := mat.NewDense(3, 3, []float64{1, 2, 3, 0, 4, 5, 0, 0, 6})
b := mat.NewDense(3, 3, []float64{8, 9, 10, 1, 4, 2, 9, 0, 2})
c := mat.NewDense(3, 2, []float64{3, 2, 1, 4, 0, 8})


    

In [3]:

    

fa := mat.Formatted(a, mat.Prefix("    "))
fb := mat.Formatted(b, mat.Prefix("    "))
fmt.Printf("\na = %0.4v\n\n", fa)
fmt.Printf("b = %0.4v\n\n", fb)


    

    
Out[3]:





a = ⎡1  2  3⎤
    ⎢0  4  5⎥
    ⎣0  0  6⎦

b = ⎡ 8   9  10⎤
    ⎢ 1   4   2⎥
    ⎣ 9   0   2⎦

In [4]:

    

// Add a and b.
d := new(mat.Dense)
d.Add(a, b)
fd := mat.Formatted(d, mat.Prefix("            "))
fmt.Printf("d = a + b = %0.4v\n\n", fd)

// Multiply a and c.
f := new(mat.Dense)
f.Mul(a, c)
ff := mat.Formatted(f, mat.Prefix("          "))
fmt.Printf("f = a c = %0.4v\n\n", ff)

// Raising a matrix to a power.
g := new(mat.Dense)
g.Pow(a, 5)
fg := mat.Formatted(g, mat.Prefix("          "))
fmt.Printf("g = a^5 = %0.4v\n\n", fg)

// Apply a function to each of the elements of a.
h := new(mat.Dense)
sqrt := func(_, _ int, v float64) float64 { return math.Sqrt(v) }
h.Apply(sqrt, a)
fh := mat.Formatted(h, mat.Prefix("              "))
fmt.Printf("h = sqrt(a) = %0.4v\n\n", fh)


    

    
Out[4]:





d = a + b = ⎡ 9  11  13⎤
            ⎢ 1   8   7⎥
            ⎣ 9   0   8⎦

f = a c = ⎡ 5  34⎤
          ⎢ 4  56⎥
          ⎣ 0  48⎦

g = a^5 = ⎡        1        682  1.074e+04⎤
          ⎢        0       1024  1.688e+04⎥
          ⎣        0          0       7776⎦

h = sqrt(a) = ⎡    1  1.414  1.732⎤
              ⎢    0      2  2.236⎥
              ⎣    0      0  2.449⎦

In [ ]: