(window_size, sigma)
| 317 | |
| 318 | |
| 319 | def gaussian(window_size, sigma): |
| 320 | gauss = torch.Tensor([exp(-(x - window_size // 2) ** 2 / float(2 * sigma ** 2)) for x in range(window_size)]) |
| 321 | return gauss / gauss.sum() |
| 322 | |
| 323 | |
| 324 | def create_window(window_size, channel): |