标签： python tensorflow
我有两个1级张量，其中一个包含X个浮点数，另一个是一个热点，也带有X个条目。
我想用一个元素创建一个新的rank-1张量：第一个张量中的浮点数，其中一个热矢量等于1。
任何帮助将不胜感激。
答案 0 :(得分：0)
听起来像tf.boolean_mask是您要寻找的东西： https://www.tensorflow.org/api_docs/python/tf/boolean_mask