
Глубокое обучение нейронных сетей на основе PyTorch
369
12.2.2. Создание тензоров в PyTorch
В этом разделе мы изучим несколько различных способов создания тензоров, а затем
рассмотрим некоторые их свойства и узнаем, как с ними обращаться. Мы можем соз-
дать тензор из простого списка или массива NumPy, используя функцию
torch.tensor
или
torch.from_numpy
:
>>> import torch
>>> import numpy as np
>>> np.set_printoptions(precision=3)
>>> a = [1, 2, 3]
>>> b = np.array([4, 5, 6], dtype=np.int32)
>>> t_a = torch.tensor(a)
>>> t_b = torch.from_numpy(b)
>>> print(t_a)
>>> print(t_b)
tensor([1, 2, 3])
tensor([4, 5, 6], dtype=torch.int32) ...