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

// 创建一个2维张量
const tensor = tf.tensor2d([[1, 2, 3], [4, 5, 6], [7, 8, 9]]);

// 读取第一行
const firstRow = tensor.slice([0, 0], [1, tensor.shape[1]]);
console.log('第一行:', firstRow.dataSync());

// 读取第二行
const secondRow = tensor.slice([1, 0], [1, tensor.shape[1]]);
console.log('第二行:', secondRow.dataSync());

// 写入新的值到第三行
const newRow = tf.tensor1d([10, 11, 12]);
const updatedTensor = tensor.slice([0, 0], [2, tensor.shape[1]]).concat(newRow.reshape([1, tensor.shape[1]]), 0);
console.log('更新后的张量:', updatedTensor.dataSync());