如何在 TensorFlow.js 中创建一个向量，其中第 i 个条目是提高到第 i 次方的标量？

How to create a vector in TensorFlow.js, where the i-th entry is a scalar raised to the i-th power?

提问人：centershocks 提问时间：9/20/2022 最后编辑：AloneTogethercentershocks 更新时间：9/20/2022 访问量：65

问：

我想在 TensorFlow.js 中创建一个向量，它具有以下结构：

$v = \left[ \begin{array}{cccccc}a^0 & a^1 & a^2 & \dots & a^{n-1} & a^n \end{array} \right]$

其中是非复标量，是向量的长度，是向量的第 -个条目，从零开始计数。anii

有没有快速的方法可以做到这一点，或者我需要循环吗？

张量流 .js

答：

1赞 AloneTogether 9/20/2022 #1

IIUC，你可以尝试这样的事情：

x = tf.tensor1d([1, 2, 3, 4, 5, 6, 7, 8]).toInt();
y = tf.range(0, x.shape[0], 1).toInt();

x.pow(y).print();

"Tensor
    [1, 2, 9, 64, 625, 7776, 117649, 2097152]"