Перечислить массив, помещенный в torch.autograd.Variable - PullRequest
0 голосов
/ 08 мая 2020

У меня есть переменная xy, которая загружает данные из файла и используется для создания x_data и y_data переменных типа torch.autograd.Variable. Я полагаю, что содержимое x_data и y_data - это массивы. Но как пронумеровать этот массив, чтобы распечатать содержимое?

d = r"C:\gdrive\python\diabetes.csv"
import torch
from torch.autograd import Variable 
import numpy as np
import torch.nn.functional as F
xy = np.loadtxt(d,delimiter=',',dtype=np.float32 )
print("***")
print(xy)
x_data = Variable(torch.from_numpy(xy[:,0:-1]))
y_data=  Variable(torch.from_numpy(xy[:,[-1]]))

Ответы [ 2 ]

0 голосов
/ 08 мая 2020

Вы должны иметь возможность перебирать тензоры pytorch точно так же, как python списки:

import torch
from torch.autograd import Variable
import numpy as np
import torch.nn.functional as F

d = r"C:\gdrive\python\diabetes.csv"
xy = np.loadtxt(d, delimiter=',', dtype=np.float32)
print("***")
x_data = Variable(torch.from_numpy(xy[:,0:-1]))
y_data = Variable(torch.from_numpy(xy[:,[-1]]))

print(x_data.shape)
print(y_data.shape)
for i, (x, y) in enumerate(zip(x_data, y_data)):
    print("example:", i)
    print("features:", x)
    print("label:", y)
    print()
0 голосов
/ 08 мая 2020

Вы можете перечислять / перебирать тензоры Pytorch так же, как для numpy массивов и стандартных Python объектов.

...