如何按元素求平方根矩阵?

时间:2018-06-22 23:24:24

标签: go matrix gonum

我本打算使用Pow,但它似乎只接受整数值的幂运算。

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

a = ⎡ 4   9⎤
    ⎣16  25⎦

m = ⎡2  3⎤
    ⎣4  5⎦

1 个答案:

答案 0 :(得分:0)

对于矩阵的逐个元素的平方根,编写如下内容:

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⎦