(self, a, b)
| 3325 | return tf.transpose(a, perm=axes) |
| 3326 | |
| 3327 | def matmul(self, a, b): |
| 3328 | return tnp.matmul(a, b) |
| 3329 | |
| 3330 | # todo(okachaiev): replace this with a more reasonable implementation |
| 3331 | def nan_to_num(self, x, copy=True, nan=0.0, posinf=None, neginf=None): |