(self, inp)
| 749 | self.horizontal = horizontal |
| 750 | |
| 751 | def forward(self, inp): |
| 752 | assert isinstance( |
| 753 | inp, Tensor |
| 754 | ), "expected input is megengine.Tensor, got {}".format(type(inp)) |
| 755 | return flip(inp, self.vertical, self.horizontal) |
| 756 | |
| 757 | |
| 758 | class RandomHorizontalFlip(Module): |