Функция tensor

Функция tensor создает тензор из переданных данных. Первым параметром функция принимает список, кортеж или массив. Вторым параметром можно передать тип данных элементов.

Синтаксис

torch.tensor(data, [dtype])

Пример

Давайте создадим тензор из списка чисел 1, 2, 3, 4, 5:

import torch

t = torch.tensor([1, 2, 3, 4, 5])
print(t)

Результат выполнения кода:

tensor([1, 2, 3, 4, 5])

Пример

Давайте создадим тензор с типом float:

import torch

t = torch.tensor([1, 2, 3, 4, 5], dtype=torch.float)
print(t)

Результат выполнения кода:

tensor([1., 2., 3., 4., 5.])

Пример

Создадим тензор из двумерного списка:

import torch

t = torch.tensor([[1, 2, 3], [4, 5, 6]])
print(t)

Результат выполнения кода:

tensor([
	[1, 2, 3],
	[4, 5, 6],
])

Пример

Используем tensor с массивом NumPy:

import torch
import numpy as np

arr = np.array([1, 2, 3, 4, 5])
t = torch.tensor(arr)
print(t)

Результат выполнения кода:

tensor([1, 2, 3, 4, 5])

Смотрите также

функцию as_tensor,
которая преобразует данные в тензор без лишнего копирования
функцию from_numpy,
которая создает тензор из массива NumPy
функцию zeros,
которая создает тензор, заполненный нулями
функцию ones,
которая создает тензор, заполненный единицами