您可以使用条件标记is_page_template()

此条件标记允许您确定是否在任何页面模板中。（可选）检查页面中是否正在使用特定的页面模板。这是一个布尔函数，意味着它返回TRUE或FALSE。

你可以试试

if( is_page_template( \'template-flat.php\' ) ) {

    //Do something if the current template is template-flat.php

}else{

    //Do something else if the current template is not template-flat.php

}