Как создать тензор в pytorch?

Пользователь

от camren , в категории: Python , 2 года назад

Как создать тензор в pytorch?

Facebook Vk Ok Twitter LinkedIn Telegram Whatsapp

2 ответа

Пользователь

от kenya , 2 года назад

@camren 

В PyTorch можно создать тензор, используя функцию torch.tensor(). Например, чтобы создать тензор размером (2, 3) с нулевыми значениями:

1
2
3
4
import torch

tensor = torch.tensor([[0, 0, 0], [0, 0, 0]])
print(tensor)


Вы можете также создать тензор с различными типами данных, используя аргумент dtype, и ини

Пользователь

от miguel_ritchie , год назад

@camren 

циализировать его значения с помощью функции torch.zeros(). Например, чтобы создать тензор размером (3, 2) из вещественных чисел:


1 2 3 4


import torch


tensor = torch.zeros(3, 2, dtype=torch.float32) print(tensor)


Если вы хотите создать случайный тензор, вы можете использовать функцию torch.rand(). Например, чтобы создать тензор размером (4, 4) со случайными значениями от 0 до 1:


1 2 3 4


import torch


tensor = torch.rand(4, 4) print(tensor)


Также вы можете создать тензор с предопределенными значениями, используя функции, такие как torch.ones() или torch.eye(). Например, чтобы создать тензор размером (2, 2) из единиц:


1 2 3 4


import torch


tensor = torch.ones(2, 2) print(tensor)


Это всего лишь несколько примеров способов создания тензоров в PyTorch. Вы также можете использовать другие функции для создания тензоров с определенной структурой или значениями. Более подробную информацию вы можете найти в документации PyTorch.