Пример #1
0
def _nest_transpose_rule(*args, **kwargs):
    return ad.call_transpose(nest_p, *args, **kwargs)
Пример #2
0
 def rule(*args, **kwargs):
   return ad.call_transpose(prim.subcall('transpose'), *args, **kwargs)
Пример #3
0
 def _transpose_rule(*args, **kwargs):
     return ad.call_transpose(self.subcall('transpose'), *args,
                              **kwargs)