TensorFlow.js tf.layers.minimum()函数

在Tensorflow.js中，tf.layers.minimum()函数被用来创建两个（或多个）张量元素的逐元素最小值的层。

语法

tf.layers.minimum(config)

参数

• config: 一个对象，包含以下属性
  • inputShape（可选）: 输入张量的尺寸。
  • name（可选）: 层的名称。

返回值

一个最小值层的实例。

示例

以下代码示例创建了一个最小值层，并将其应用于一个输入张量。

const input = tf.input({shape: [3]});
const minLayer = tf.layers.minimum();
const output = minLayer.apply(input);

console.log(`Layer output shape: ${JSON.stringify(output.shape)}`);

输出

Layer output shape: [null,3]

注意事项

• 如果您在创建模型时使用此层，则不需要提供输入张量的形状，因为它将由模型自动计算。
• 此层支持常规的Keras层参数，例如kernelInitializer和biasInitializer。
• 如果此层的输入张量具有大于2的rank，则该函数将对其前两个轴执行逐元素最小值操作。

参考

• tf.layers.minimum() TensorFlow.js官方文档