Как я могу построить тензор Pytorch? ⇐ Python

[Anonymous]

Я хотел бы построить тензор графического процессора pytorch:

Код: Выделить всё

input= torch.randn(100).to(device)
output = torch.where(input>=0, input, -input)

input = input.('cpu').detach().numpy().copy()
output = output.('cpu').detach().numpy().copy()

plt.plot(input,out)

Однако я пытаюсь преобразовать эти тензоры в процессор, numpy, это не работает. Как я могу построить тензоры?

Подробнее здесь: https://stackoverflow.com/questions/686 ... rch-tensor