lzhengning
/
JittorMirror
mirror of https://github.com/Jittor/Jittor
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

add numel

This commit is contained in:
Dun Liang 2020-06-15 16:42:07 +08:00
parent 17d38e723a
commit 0c57b4e44b
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/var_holder.h
View File

@ -75,6 +75,7 @@ struct VarHolder {
return var->shape[dim];
}
// @pyjt(numel)
inline int64 numel() {
if (var->num<0) sync();
return var->num;