我如何逐元素平方根 gonum 矩阵？

Question

我打算使用 Pow，但它似乎只接受整数值。

例如矩阵a的逐元素平方根矩阵m.:

a = ⎡ 4   9⎤
    ⎣16  25⎦

m = ⎡2  3⎤
    ⎣4  5⎦

Answer 1

对于矩阵的逐元素平方根，写这样的东西：

package main

import (
    "fmt"
    "math"

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

func main() {
    a := mat.NewDense(2, 2, []float64{
        4, 9,
        16, 25,
    })
    fa := mat.Formatted(a, mat.Prefix("    "), mat.Squeeze())
    fmt.Printf("a = %v\n\n", fa)

    m := new(mat.Dense)
    m.Apply(func(i, j int, v float64) float64 { return math.Sqrt(v) }, a)
    fm := mat.Formatted(m, mat.Prefix("    "), mat.Squeeze())
    fmt.Printf("m = %v\n\n", fm)
}

输出：

a = ⎡ 4   9⎤
    ⎣16  25⎦

m = ⎡2  3⎤
    ⎣4  5⎦

我如何逐元素平方根 gonum 矩阵？

How do I element-wise square root a gonum matrix?

matrix

go

gonum