TensorflowJS：如何在张量的特定索引处读取和写入一行？

Question

在 Tensorflow JS 中，如何在张量的特定索引处读取和写入行？ 我想做这样的事情：

let a = tf.tensor([[1, 2], [3, 4]]);
let b = tf.tensor([[5, 6], [7, 8]]);
a[0] -= b[1]
a.print()
b.print()

这应该打印：

[[-4, -4],
[3, 4]]

[[5, 6]
[7, 8]]

另外，我在哪里可以找到如何对 javascript 张量进行其他基本操作？ （到目前为止我只发现了这个： https://js.tensorflow.org/api/0.6.1/#class:Tensor ）

编辑：这回答了我的问题的阅读部分： 如何在 TensorFlow 中获取张量的特定行？

Answer 1

张量是不可变的，因此在创建后无法更改其值。 可以做的是创建一个新的张量，并给它张量 a 和 b 的值。

let a = tf.tensor([[1, 2], [3, 4]]);
let b = tf.tensor([[5, 6], [7, 8]]);

tf.concat([a.sub(b).slice(1, 1), a.slice(1,1)]).print()

// or

const cond = tf.tensor1d([false, true], 'bool');

a.sub(b).where(cond, a).print()