def gaussian(window_size, sigma):
    gauss = torch.exp(torch.Tensor([((- ((x - (window_size // 2)) ** 2)) / float((2 * (sigma ** 2)))) for x in range(window_size)]).float())
    return (gauss / gauss.sum())
def gaussian(window_size, sigma):
    gauss = torch.exp(torch.Tensor([((- ((x - (window_size // 2)) ** 2)) / float((2 * (sigma ** 2)))) for x in range(window_size)]).float())
    return (gauss / gauss.sum())