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()