torch.tensor

torch.tensor(data, *, dtype=None, device=None, requires_grad=False, pin_memory=False) → Tensor

Создаёт тензор без истории autograd (также известный как «листовой тензор», см. Механика autograd), копируя data.

Предупреждение

При работе с тензорами для большей наглядности используйте torch.Tensor.clone(), torch.Tensor.detach() и torch.Tensor.requires_grad_(). Если t — это тензор, то torch.tensor(t) эквивалентно t.detach().clone(), а torch.tensor(t, requires_grad=True) эквивалентно t.detach().clone().requires_grad_(True).

См. также

torch.as_tensor() сохраняет историю autograd и по возможности избегает копирования. torch.from_numpy() создаёт тензор, использующий общее хранилище с массивом NumPy.

Параметры:

data (array_like) – Исходные данные для тензора. Это может быть список, кортеж, массив NumPy ndarray, скаляр или данные других типов.

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

• dtype (torch.dtype, необязательно) – желаемый тип данных возвращаемого тензора. По умолчанию: если None, тип данных определяется по data.
• device (torch.device, необязательно) – устройство, на котором создаётся тензор. Если значение равно None и data является тензором, используется устройство data. Если значение равно None и data не является тензором, результирующий тензор создаётся на текущем устройстве.
• requires_grad (bool, необязательно) – Нужно ли autograd записывать операции над возвращаемым тензором. По умолчанию: False.
• pin_memory (bool, необязательно) – Если задано, возвращаемый тензор будет размещён в закреплённой памяти. Работает только для тензоров CPU. По умолчанию: False.

Пример:

>>> torch.tensor([[0.1, 1.2], [2.2, 3.1], [4.9, 5.2]])
tensor([[ 0.1000,  1.2000],
        [ 2.2000,  3.1000],
        [ 4.9000,  5.2000]])

>>> torch.tensor([0, 1])  # Type inference on data
tensor([ 0,  1])

>>> torch.tensor([[0.11111, 0.222222, 0.3333333]],
...              dtype=torch.float64,
...              device=torch.device('cuda:0'))  # creates a double tensor on a CUDA device
tensor([[ 0.1111,  0.2222,  0.3333]], dtype=torch.float64, device='cuda:0')

>>> torch.tensor(3.14159)  # Create a zero-dimensional (scalar) tensor
tensor(3.1416)

>>> torch.tensor([])  # Create an empty tensor (of size (0,))
tensor([])