(self)
| 72 | self.assertTupleEqual(result.shape, expected_shape) |
| 73 | |
| 74 | def test_additional(self): |
| 75 | for p in TEST_NDARRAYS: |
| 76 | out = AsDiscrete(argmax=True, dim=1, keepdim=False)(p([[[0.0, 1.0]], [[2.0, 3.0]]])) |
| 77 | assert_allclose(out, p([[0.0, 0.0], [0.0, 0.0]]), type_test=False) |
| 78 | |
| 79 | |
| 80 | if __name__ == "__main__": |
nothing calls this directly
no test coverage detected