import torchimport torchnn as nnimport torchoptim as optimimport pandas as pdfrom sklearnpreprocessing import StandardScaler# 读入Excel表格data = pdread_excelCUserslenovoDesktopHIVDNN神经网络测试data1xlsxx = da
这个错误通常是由于在调用torch.max()函数时,维度超出了范围所导致的。在这里,可能是因为labels的维度不是1,而是2,导致在torch.max()中指定维度为1时出现了错误。可以尝试将labels的维度转换为1维,例如使用labels = torch.tensor(labels).squeeze().long()来解决这个问题。
原文地址: https://www.cveoy.top/t/topic/bcAi 著作权归作者所有。请勿转载和采集!