枚举数组放置在torch.autograd.Variable中

问题描述 投票:0回答:2

我有变量xy,它从文件加载数据,并用于创建x_data类型的y_datatorch.autograd.Variable变量。我想x_datay_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]]))
python pytorch
2个回答
0
投票

您可以像对numpy数组和标准Python对象一样,枚举/迭代Pytorch张量。


0
投票

您应该能够像python列表一样迭代pytorch张量:

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("featurers:", x)
    print("label:", y)
    print()
© www.soinside.com 2019 - 2024. All rights reserved.