我想通过Tensorflow实现以下目标（最好不要使用 tf.feature_column.bucketized_column）：
给出一个我将定义的存储桶，例如： boundary = tf.constant（[-1，-0.8，-0.6，-0.4，-0.2,0,0.2,0.4,0.6,0.8,1]）
存储桶[-inf，1]的ID为0，存储桶[-1，-0.8]的ID为1，依此类推。
在图的某个点上，我将得到张量K，其值为[0.1，-10，-0.1] 并且我想构建一个函数，该函数将返回该数字所属的存储桶的ID：
所以K-> [6,0,5]
有什么建议吗？