o Ne@sdZddlZddlmmZddlmZdZ dZ dZ dZ dZ dd d Zd d Zdd dZdddZddZ    dddZddZdS)zUtils for preprocessing layers.N)tf_utilsintZone_hotZ multi_hotcountZtf_idfcCsDt|tjtjtjfst||}|dur |j|kr t||}|S)zs(     =