import torch


def ones(shape, dtype=None):
    """Wrapper of `torch.ones`.

    Parameters
    ----------
    shape : tuple of ints
        Shape of output tensor.
    dtype : data-type, optional
        Data type of output tensor, by default None
    """
    return torch.ones(shape, dtype=dtype)