TensorFlow.js tf.mod()函数

TensorFlow.js 的 tf.mod() 函数用于对张量进行取模运算，即张量中的每一个元素对给定的除数取余数。

语法

tf.mod(x, y)

参数说明

• x：一个张量，必选参数。
• y：一个张量，必选参数。用于进行取模运算的除数张量。两个张量的形状必须相同。

返回值

返回一个张量，与输入张量的形状一致，其中每个元素都是输入张量对给定除数的余数。

代码示例

const x = tf.tensor1d([5, 7, 9, 11]);
const y = tf.tensor1d([2, 3, 4, 5]);
const result = tf.mod(x, y);

result.print();

输出结果为：[1, 1, 1, 1]

上述代码中，我们首先创建了两个一维张量 x 和 y，分别表示被除数和除数。然后，我们使用 tf.mod() 函数将两个张量进行取模运算，得到对应的余数张量 result。最后，我们通过 result.print() 将余数张量输出到控制台。