如何逐元素平方根 gonum 矩阵?

Question

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

例如，矩阵的逐元素平方根矩阵ma：

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 = %vnn", 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 = %vnn", fm)
}

输出：

a = ⎡ 4   9⎤
⎣16  25⎦
m = ⎡2  3⎤
⎣4  5⎦