// MatrixComplexMultiSimple is an operation that will multiple two complex matrices of any size together
func MatrixComplexMultiSimple(matrixA ct.MatrixComplex, matrixB ct.MatrixComplex) (ct.MatrixComplex, error) {
if matrixA.GetNumCols() != matrixB.GetNumRows() {
return nil, errors.New("Length of columns of matrix A not equal to length of rows of matrix B")
}
if matrixA.IsIdentity() {
return matrixB, nil
}
if matrixB.IsIdentity() {
return matrixA, nil
}
var sum complex128
matrixAB := ct.MakeComplexMatrix(matrixA.GetNumRows(), matrixB.GetNumRows())
for i := 0; i < matrixA.GetNumRows(); i++ {
for j := 0; j < matrixB.GetNumCols(); j++ {
sum = 0.0
for k := 0; k < matrixA.GetNumCols(); k++ {
sum += matrixA.Get(i, k) * matrixB.Get(k, j)
}
matrixAB.Set(i, j, sum)
}
}
return matrixAB, nil
}
// MatrixComplexSubtraction is an operation that will subtract two complex matrices from one another
func MatrixComplexSubtraction(matrixA ct.MatrixComplex, matrixB ct.MatrixComplex) (ct.MatrixComplex, error) {
if matrixA.GetNumCols() != matrixB.GetNumCols() && matrixA.GetNumRows() != matrixB.GetNumRows() {
return nil, errors.New("Matrices do not have equivalent dimensions")
}
matrixAB := ct.MakeComplexMatrix(matrixA.GetNumRows(), matrixB.GetNumCols())
for i := 0; i < matrixA.GetNumRows(); i++ {
for j := 0; j < matrixA.GetNumCols(); j++ {
matrixAB.Set(i, j, matrixA.Get(i, j)-matrixB.Get(i, j))
}
}
return matrixAB, nil
}
// OuterProductComplex returns the outer product for real Vectors
func OuterProductComplex(vectorA ct.VectorComplex, vectorB ct.VectorComplex) (ct.MatrixComplex, error) {
if vectorA.Type() != ct.ColVector || vectorB.Type() != ct.RowVector {
return nil, errors.New("One or both vector types are not consistent with the vector inner product")
}
matrix := ct.MakeComplexMatrix(vectorA.Dim(), vectorB.Dim())
for i := 0; i < vectorA.Dim(); i++ {
for j := 0; j < vectorB.Dim(); j++ {
matrix.Set(i, j, vectorA.Get(i)*vectorB.Get(j))
}
}
return matrix, nil
}
