(self, a: MetaTensor, b: MetaTensor)
| 67 | comp(id(a), id(b)) |
| 68 | |
| 69 | def check_meta(self, a: MetaTensor, b: MetaTensor) -> None: |
| 70 | self.assertEqual(a.is_batch, b.is_batch) |
| 71 | meta_a, meta_b = a.meta, b.meta |
| 72 | # need to split affine from rest of metadata |
| 73 | aff_a = meta_a.get("affine", None) |
| 74 | aff_b = meta_b.get("affine", None) |
| 75 | assert_allclose(aff_a, aff_b) |
| 76 | meta_a = {k: v for k, v in meta_a.items() if k != "affine"} |
| 77 | meta_b = {k: v for k, v in meta_b.items() if k != "affine"} |
| 78 | self.assertEqual(meta_a, meta_b) |
| 79 | |
| 80 | def check( |
| 81 | self, |
no test coverage detected