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

remove matmul print

This commit is contained in:
li-xl 2020-10-21 11:36:59 +08:00
parent 84c10c7a06
commit bd506460be
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
python/jittor/nn.py
View File

@ -118,7 +118,7 @@ Example::
# TODO:ugly implementation for tuner # TODO:ugly implementation for tuner
aa = a.reshape((-1, m)) aa = a.reshape((-1, m))
cc = matmul(aa, b) cc = matmul(aa, b)
print(a.shape, b.shape, cc.shape) # print(a.shape, b.shape, cc.shape)
return cc.reshape(a.shape[:-1] + [k]) return cc.reshape(a.shape[:-1] + [k])
for i in range(len_c-2): for i in range(len_c-2):
ai = len_a-(len_c-i) ai = len_a-(len_c-i)