我想知道是否有一個(gè)包可以在 Go 中提供高效的逐元素矩陣運(yùn)算？類似于 GSL 的東西？
2 回答
德瑪西亞99
TA貢獻(xiàn)1770條經(jīng)驗(yàn) 獲得超3個(gè)贊
通過 cgo 調(diào)用例如 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)
}
慕森王
TA貢獻(xiàn)1777條經(jīng)驗(yàn) 獲得超3個(gè)贊
GSL 有各種 cgo 綁定，甚至還有一些對純 Go 端口的嘗試。似乎還沒有多少人認(rèn)可（就明星而言）并且已經(jīng)有幾個(gè)月沒有活動了，但您可能想看看代碼：
- 2 回答
- 0 關(guān)注
- 243 瀏覽
添加回答
舉報(bào)
0/150
提交
取消