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

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

от camren , в категории: Python , 5 месяцев назад

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

Facebook Vk Ok Twitter LinkedIn Telegram Whatsapp

1 ответ

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

от kenya , 5 месяцев назад

@camren 

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

1
2
3
4
import torch

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


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