TensorFlow中如何对连续型序列特征进行Embedding？

比如一个序列特征的维度是：
[batch_size, k]
我想序列中的每一个元素先进行OneHot，使用给定的分割点boundaries，类似于bucketized_column那样，得到：
[batch_size, k, n]
最后进行Embedding，设定输出的维度是M：
[batch_size, k, m]