(self, a, b, should_match)
| 63 | return m, t |
| 64 | |
| 65 | def check_ids(self, a, b, should_match): |
| 66 | comp = self.assertEqual if should_match else self.assertNotEqual |
| 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) |