import torch


def init_method_uniform(low, high):
    """Init method based on Uniform(low, high)."""
    def init_(tensor):
        return torch.nn.init.uniform_(tensor, a=low, b=high)
    return init_