numpy.ndarray 转 pytorch Tensor

特征 ndarray 转 tensor torch.Tensor(train_examples[0][0])报错：
TypeError: new(): data must be a sequence (got numpy.float64)

需要改成：
torch.Tensor(train_examples[0][0].reshape(1, n_feature))