mirror of https://github.com/Jittor/Jittor
delect all delete any
This commit is contained in:
parent
32af287461
commit
0efca41bba
|
@ -104,18 +104,6 @@ def t(x):
|
|||
return x.transpose(*pose)
|
||||
jt.Var.t = t
|
||||
|
||||
def any(x,dim=0,keepdim=False):
|
||||
y = x.int().sum(dim,keepdims=keepdim)
|
||||
return y>0
|
||||
jt.Var.any = any
|
||||
|
||||
def all(x,dim=0,keepdim=False):
|
||||
y = x.int().sum(dim,keepdims=keepdim)
|
||||
return y==x.shape[dim]
|
||||
|
||||
jt.Var.all = all
|
||||
|
||||
|
||||
def median(x,dim=None,keepdim=False):
|
||||
if dim is None:
|
||||
x = x.reshape(-1)
|
||||
|
|
Loading…
Reference in New Issue