TensorflowJS：如何读取和写入 Tensor 特定索引处的行？

Question

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

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）

编辑：这回答了我问题的已读部分：How can I get specific rows of a tensor in 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()

TensorflowJS：如何读取和写入 Tensor 特定索引处的行？

TensorflowJS: How can I read and write to a row at a specific index of a Tensor?

javascript

tensorflow

tensorflow.js