torch.nn获取输出层对象