pytorch基本流程

数据集的预处理模块 tensor格式转换 主要是讲numpy格式转换为tensor格式 x_train, y_train, x_valid, y_valid = map( torch.tensor, (x_train, y_train, x_valid, y_valid) ) torch.nn.finction模块与torch.nn.Module模块 torch.nn.functional中有很多功能,后续会常用的。那什么时候使用nn.Module,什么时候使用nn.functional呢?一般情况下,如果模型有可学习的参数,最好用nn.Module,其他情况nn.functional相对更简