| 7 | |
| 8 | |
| 9 | class StaticDynamicAxis(object): |
| 10 | def __init__(self, static, dynamic): |
| 11 | self.static = static |
| 12 | self.dynamic = dynamic |
| 13 | |
| 14 | def apply(self, f): |
| 15 | try: |
| 16 | st = f(self.static) |
| 17 | return StaticDynamicAxis(st, st) |
| 18 | except TypeError: |
| 19 | return StaticDynamicAxis(None, f(self.dynamic)) |
| 20 | |
| 21 | def __str__(self): |
| 22 | return "S={}, D={}".format(str(self.static), str(self.dynamic)) |
| 23 | |
| 24 | |
| 25 | def DynamicLazyAxis(shape, idx): |