torch.ones

torch.ones(*size, *, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False) → Tensor

Возвращает тензор, заполненный скалярным значением 1, с формой, заданной переменным аргументом size.

Параметры:

size (int...) – последовательность целых чисел, задающая форму выходного тензора. Может быть переменным числом аргументов или коллекцией, например списком или кортежем.

Именованные аргументы:

• out (Tensor, optional) – выходной тензор.
• dtype (torch.dtype, optional) – требуемый тип данных возвращаемого тензора. По умолчанию: если None, используется глобальный тип по умолчанию (см. torch.set_default_dtype()).
• layout (torch.layout, optional) – требуемый формат возвращаемого тензора. По умолчанию: torch.strided.
• device (torch.device, optional) – требуемое устройство для возвращаемого тензора. По умолчанию: если None, используется текущее устройство для типа тензора по умолчанию (см. torch.set_default_device()). device будет CPU для типов тензоров CPU и текущим устройством CUDA для типов тензоров CUDA.
• requires_grad (bool, optional) – следует ли autograd записывать операции над возвращаемым тензором. По умолчанию: False.

Пример:

>>> torch.ones(2, 3)
tensor([[ 1.,  1.,  1.],
        [ 1.,  1.,  1.]])

>>> torch.ones(5)
tensor([ 1.,  1.,  1.,  1.,  1.])