擅长:python、mysql、java
<p>也许可以试试看它是否可以解决您的问题(基于您的示例代码)?</p>
<pre><code>train_target = torch.tensor(train['Target'].values.astype(np.float32))
train = torch.tensor(train.drop('Target', axis = 1).values.astype(np.float32))
train_tensor = data_utils.TensorDataset(train, train_target)
train_loader = data_utils.DataLoader(dataset = train_tensor, batch_size = batch_size, shuffle = True)
</code></pre>