(dim: Optional[int])
| 9162 | self.checkScript(norm, ()) |
| 9163 | |
| 9164 | def torch_unique(dim: Optional[int]): |
| 9165 | ten = torch.unique(torch.tensor([[1, 3], [2, 3]], dtype=torch.long)) |
| 9166 | a = torch.unique(ten, dim=dim) |
| 9167 | b = torch.unique(ten, return_counts=True, dim=dim) |
| 9168 | c = torch.unique(ten, return_inverse=True, dim=dim) |
| 9169 | d = torch.unique(ten, return_counts=True, return_inverse=True, dim=dim) |
| 9170 | return a, b, c, d |
| 9171 | |
| 9172 | self.checkScript(torch_unique, (None,)) |
| 9173 | self.checkScript(torch_unique, (0,)) |