如何在TensorFlow.js中创建第i项为标量a的i次幂的向量？

完全不需要循环，直接用TensorFlow.js的内置张量操作就能高效实现——这类批量运算基于GPU加速，比循环的性能好得多。

实现步骤 #

• 先生成从0到n的整数序列张量（对应每个元素的指数）
• 借助tf.pow的广播特性，计算标量a的对应幂次

示例代码 #

如果n是向量最后一项的指数（向量长度为n+1）：

// 定义参数
const a = 2; // 非复标量
const n = 5; // 最后一项的指数

// 生成目标向量
const v = tf.pow(a, tf.range(0, n + 1, 1));

// 验证结果
v.print();
// 输出：[1, 2, 4, 8, 16, 32]

如果n指的是向量的总长度（即最后一项的指数为n-1），只需调整tf.range的结束值：

const n = 6; // 向量长度
const v = tf.pow(a, tf.range(0, n, 1));
// 输出同样为[1, 2, 4, 8, 16, 32]

原理说明 #

• tf.range(0, end, step)会生成一个从起始值到end（不包含end）的整数序列张量，步长设为1刚好覆盖所有需要的指数。
• tf.pow支持广播机制：当第一个参数是标量、第二个是张量时，会自动将标量广播到和张量相同的形状，然后逐元素计算幂次，完美匹配需求。

内容的提问来源于stack exchange，提问作者centershocks