pytorch修改层类型