Как поэлементно получить квадратный корень из матрицы гонумов?

Я собирался использовать Pow но кажется, что он принимает питание только от целочисленных значений.

Например, поэлементная квадратно-корневая матрица m матрицы a.:

a = ⎡ 4   9⎤
    ⎣16  25⎦

m = ⎡2  3⎤
    ⎣4  5⎦

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⎦
Другие вопросы по тегам