如何将矩阵与 gonum 中的向量相乘?

Question

我想将mat.Dense矩阵与mat.VecDense向量相乘，但显然mat.Dense也没有mat.VecDens实现矩阵接口或定义将矩阵与向量相乘的方法。我该怎么做？

Answer 1

解决了。

mat.NewVecDense(...)返回一个*mat.VecDense，该实现一个方法func MulVec(a mat.Matrix, b mat.Vector)

这是验证功能的测试

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)
}