andr3/gotch
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fixed lstsq API

Browse Source
This commit is contained in:
sugarme 2021-05-19 13:59:21 +10:00
parent 0d86ab1cf3
commit 367f57fdea
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensor/patch.go
View File

@ -565,7 +565,7 @@ func MustWhere(condition Tensor, del bool) (retVal []Tensor) {
// ====================================================================
// void atg_lstsq(tensor *, tensor self, tensor A);
func (ts *Tensor) Lstsq(a *ts.Tensor, del bool) (retVal *Tensor, err error) {
func (ts *Tensor) Lstsq(a *Tensor, del bool) (retVal *Tensor, err error) {
if del {
defer ts.MustDrop()
}
@ -580,7 +580,7 @@ func (ts *Tensor) Lstsq(a *ts.Tensor, del bool) (retVal *Tensor, err error) {
return retVal, err
}
func (ts *Tensor) MustLstsq(a *ts.Tensor, del bool) (retVal *Tensor) {
func (ts *Tensor) MustLstsq(a *Tensor, del bool) (retVal *Tensor) {
retVal, err := ts.Lstsq(del)
if err != nil {
log.Fatal(err)