我想將mat.Dense矩陣與mat.VecDense向量相乘，但顯然mat.Dense 也mat.VecDens沒(méi)有實(shí)現(xiàn)矩陣接口或定義將矩陣與向量相乘的方法。我該怎么做？
1 回答
BIG陽(yáng)
TA貢獻(xiàn)1859條經(jīng)驗(yàn) 獲得超6個(gè)贊
解決了。
mat.NewVecDense(...)返回一個(gè)*mat.VecDense，它實(shí)現(xiàn)了一個(gè)方法func MulVec(a mat.Matrix, b mat.Vector)
這是驗(yàn)證功能的測(cè)試
func TestMatrixVectorMul(t *testing.T) {
a := mat.NewDense(3, 3, []float64{
1, 2, 3, 4, 5, 6, 7, 8, 9,
})
b := mat.NewVecDense(3, []float64{
1, 2, 3,
})
actual := make([]float64, 3)
c := mat.NewVecDense(3, actual)
// this was the method, I was looking for.
c.MulVec(a, b)
expected := []float64{14, 32, 50}
assert.Equal(t, expected, actual)
}
- 1 回答
- 0 關(guān)注
- 170 瀏覽
添加回答
舉報(bào)
0/150
提交
取消