在TensorFlow中更改张量的内容

Question

在我继续之前，请原谅我的无知。我之前有一些编程经验，但我以前的直觉让我失望了。

基本上，我需要将数字范围从0 ... K的1-D向量（大小为M x 1）扩展为2-D矩阵（或Tensor，大小为M x K），其中每行为1 1-D向量（大小为1 x K），除初始值的索引为1外，每个元素为0。

是的，这是ML类的多类分类问题。

我想到了创建一个正确形状的零矩阵，然后手动将我需要的元素的索引分配给1，但似乎无法更改已创建的变量的值。我收到错误：

TypeError: 'Tensor' object does not support item assignment

有人可以协助吗？如果您觉得我创建这个最终Tensor的方式可能会采用不同的方法，那么任何建议都会受到赞赏。

Answer 1

在张量流中，函数tf.one_hot()就是您所寻求的。 一个热门编码是描述您要实施的操作的术语。请参阅https://www.tensorflow.org/api_docs/python/tf/one_hot。