當前位置: 首頁>>代碼示例>>Golang>>正文


Golang FloatMatrix.Div方法代碼示例

本文整理匯總了Golang中github.com/henrylee2cn/algorithm/matrix.FloatMatrix.Div方法的典型用法代碼示例。如果您正苦於以下問題:Golang FloatMatrix.Div方法的具體用法?Golang FloatMatrix.Div怎麽用?Golang FloatMatrix.Div使用的例子?那麽, 這裏精選的方法代碼示例或許可以為您提供幫助。您也可以進一步了解該方法所在github.com/henrylee2cn/algorithm/matrix.FloatMatrix的用法示例。


在下文中一共展示了FloatMatrix.Div方法的1個代碼示例,這些例子默認根據受歡迎程度排序。您可以為喜歡或者感覺有用的代碼點讚,您的評價將有助於係統推薦出更棒的Golang代碼示例。

示例1: SolveDiag

/*
 * Compute
 *   X = B*diag(D).-1      flags & RIGHT == true
 *   X = diag(D).-1*C      flags & LEFT  == true
 *
 * Arguments:
 *   B     M-by-N matrix if flags&RIGHT == true or N-by-M matrix if flags&LEFT == true
 *
 *   D     N element column or row vector or N-by-N matrix
 *
 *   flags Indicator bits, LEFT or RIGHT
 */
func SolveDiag(B, D *matrix.FloatMatrix, flags Flags) {
	var c, d0 matrix.FloatMatrix
	if D.Cols() == 1 {
		// diagonal is column vector
		switch flags & (LEFT | RIGHT) {
		case LEFT:
			// scale rows; for each column element-wise multiply with D-vector
			for k := 0; k < B.Cols(); k++ {
				B.SubMatrix(&c, 0, k, B.Rows(), 1)
				c.Div(D)
			}
		case RIGHT:
			// scale columns
			for k := 0; k < B.Cols(); k++ {
				B.SubMatrix(&c, 0, k, B.Rows(), 1)
				// scale the column
				c.Scale(1.0 / D.GetAt(k, 0))
			}
		}
	} else {
		var d *matrix.FloatMatrix
		if D.Rows() == 1 {
			d = D
		} else {
			D.SubMatrix(&d0, 0, 0, 1, D.Cols(), D.LeadingIndex()+1)
			d = &d0
		}
		switch flags & (LEFT | RIGHT) {
		case LEFT:
			for k := 0; k < B.Rows(); k++ {
				B.SubMatrix(&c, k, 0, 1, B.Cols())
				// scale the row
				c.Scale(1.0 / d.GetAt(0, k))
			}
		case RIGHT:
			// scale columns
			for k := 0; k < B.Cols(); k++ {
				B.SubMatrix(&c, 0, k, B.Rows(), 1)
				// scale the column
				c.Scale(1.0 / d.GetAt(0, k))
			}
		}
	}
}
開發者ID:sguzwf,項目名稱:algorithm,代碼行數:56,代碼來源:diag.go


注:本文中的github.com/henrylee2cn/algorithm/matrix.FloatMatrix.Div方法示例由純淨天空整理自Github/MSDocs等開源代碼及文檔管理平台,相關代碼片段篩選自各路編程大神貢獻的開源項目,源碼版權歸原作者所有,傳播和使用請參考對應項目的License;未經允許,請勿轉載。