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))
·
特征 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))

更多推荐


所有评论(0)