Javascript 在tensorflow.js中设置权重的函数初始值设定项

我试图在tensorflow.js中仅使用正值初始化权重，但似乎从未给出“正确”的形状。这是我的密码：

let data_size = 500;
let input = [];
let output;
const model = tf.sequential();

for (var i = 0; i < data_size; i++){
    input[i] = i;
}

input = tf.tensor2d(input, [data_size, 1]);
output = tf.add(tf.scalar(1), input);

model.add(tf.layers.dense({units: 6, activation: "relu", inputShape: [1], weights: tf.randomUniform([6, 1], 0, 1)}));
model.add(tf.layers.dense({units: 1, activation: "linear"}));

model.compile({loss: "meanSquaredError", optimizer: "adam"});

let data_size=500；
让输入=[]；
让输出；
const model=tf.sequential（）；
对于（变量i=0；i

所以在我的代码中，在我添加的第一层中，我放置了“weights”参数来选择权重的初始化

---

但即使重量的形状是[6,1]，它也不能接受。我也尝试了tf.randomUniform（[1]，0，1），因为它可能是传递给所有权重的单个表达式，但它也不起作用。如何使用tensorflow.js选择表达式来初始化权重？

权重

是一个根据表达式的张量数组。层以这种方式初始化AX+B。因此需要提供A和B张量（其中X是层的输入）

let data_size=500；
让输入=[]；
让输出；
const model=tf.sequential（）；
对于（变量i=0；i