Tensorflow.js tf.layers getWeights()用法及代码示例

Tensorflow.js是Google开发的开源库，用于在浏览器或节点环境中运行机器学习模型和深度学习神经网络。它还可以帮助开发人员使用JavaScript语言开发ML模型，并可以直接在浏览器或Node.js中使用ML。

tf.layers.Layer.getWeights() 函数用于获取张量的权重值。

用法:

getWeights( trainableOnly? )

参数：

trainableOnly(boolean):如果为 true，该函数将仅返回可训练的权重值。

返回值：它返回一个 tf.Tensor

范例1：

Javascript


// Creating a model
const model = tf.sequential();
  
// Adding layers
model.add(tf.layers.dense({units:2, inputShape:[5]}));
model.add(tf.layers.dense({units:3}));
  
model.compile({loss:'categoricalCrossentropy', optimizer:'sgd'});
  
// Printing the weights of the layers
model.layers[0].getWeights()[0].print()
model.layers[0].getWeights()[1].print()

输出：

Tensor
    [[-0.4756567, 0.2925433 ],
     [0.3505997 , -0.5043278],
     [0.5344347 , 0.2662918 ],
     [-0.1357223, 0.2435055 ],
     [-0.6059403, 0.1990891 ]]
Tensor
    [0, 0]

范例2：

Javascript


const tf = require("@tensorflow/tfjs")
  
// Creating a model
const model = tf.sequential();
  
// Adding layers
model.add(tf.layers.dense({units:1, inputShape:[10]}));
model.add(tf.layers.dense({units:3}));
  
// Setting new weights
model.layers[0].setWeights([tf.zeros([10, 1]), tf.ones([1])]);
  
model.compile({loss:'categoricalCrossentropy', optimizer:'sgd'});
  
// Printing the weights of the layers
model.layers[0].getWeights()[0].print()
model.layers[0].getWeights()[1].print()

输出：

Tensor
    [[0],
     [0],
     [0],
     [0],
     [0],
     [0],
     [0],
     [0],
     [0],
     [0]]
Tensor
    [1]

参考：https://js.tensorflow.org/api/latest/#tf.layers.Layer.getWeights

相关用法

注：本文由纯净天空筛选整理自parasmadan15大神的英文原创作品 Tensorflow.js tf.layers getWeights() Method。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。