元素矩阵运算 Google Go？

Question

我想知道是否有一个包可以在 Go 中提供高效的逐元素矩阵运算？类似于 GSL 的东西？

Answer 1

通过 cgo 调用例如 cblas 很容易：

package main

// #include <cblas.h>

// #cgo LDFLAGS: -L/usr/lib64/atlas -lcblas

import "C"

import "fmt"

type matrix struct {

    rows  int

    cols  int

    elems []float32

}

func (a matrix) cblasmul(b matrix) (c matrix) {

    c = matrix{a.rows, b.cols, make([]float32, a.rows*b.cols)}

    C.cblas_sgemm(

        C.CblasRowMajor, C.CblasNoTrans, C.CblasNoTrans,

        C.int(a.rows), C.int(b.cols), C.int(a.cols),

        1.0,

        (*C.float)(&a.elems[0]), C.int(a.cols),

        (*C.float)(&b.elems[0]), C.int(b.cols),

        0.0,

        (*C.float)(&c.elems[0]), C.int(c.cols))

    return c

}

func main() {

    a := matrix{100, 100, make([]float32, 100*100)}

    b := matrix{100, 100, make([]float32, 100*100)}

    // ...

    c := a.cblasmul(b)

    fmt.Println(c)

}

Answer 2

GSL 有各种 cgo 绑定，甚至还有一些对纯 Go 端口的尝试。似乎还没有多少人认可（就明星而言）并且已经有几个月没有活动了，但您可能想看看代码：

http://godoc.org/?q=gsl