(window_size, sigma)
| 20 | |
| 21 | |
| 22 | def gaussian(window_size, sigma): |
| 23 | gauss = torch.Tensor([exp(-(x - window_size // 2) ** 2 / float(2 * sigma ** 2)) for x in range(window_size)]) |
| 24 | return gauss / gauss.sum() |
| 25 | |
| 26 | |
| 27 | def create_window(window_size, channel): |