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

fixed read npy failed with single element tensor with zero shape

Browse Source
This commit is contained in:
sugarme 2023-07-25 13:58:19 +10:00
parent 30068c6b41
commit 3959bc3f93
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
ts/npy.go
View File

@ -306,6 +306,11 @@ func ReadNpy(filepath string) (*Tensor, error) {
return nil, err
}
// NOTE(TT.). case tensor 1 element with shape = []
if len(data) > 0 && len(header.shape) == 0 {
header.shape = []int64{1}
}
return OfDataSize(data, header.shape, header.descr)
}
@ -348,6 +353,11 @@ func ReadNpz(filePath string) ([]NamedTensor, error) {
return nil, err
}
// NOTE(TT.). case tensor 1 element with shape = []
if len(data) > 0 && len(header.shape) == 0 {
header.shape = []int64{1}
}
tensor, err := OfDataSize(data, header.shape, header.descr)
if err != nil {
return nil, err