(self, x: torch.Tensor, y: torch.Tensor, c: float)
| 119 | super(ModuleFloatAddWithAlpha, self).__init__() |
| 120 | |
| 121 | def forward(self, x: torch.Tensor, y: torch.Tensor, c: float): |
| 122 | return torch.add(x, y, alpha=c) |
| 123 | |
| 124 | def get_random_inputs(self): |
| 125 | return (torch.randn(2, 2), torch.randn(2, 2), random.random()) |