我有一个函数,它存储张量的所有值,然后将其连接成一个矩阵。该函数如下所示:
if (i == 1) {
var y_pred1 = y_pred
} else if (i == 2) {
var y_pred2 = y_pred
} else if (i == 3) {
var y_pred3 = y_pred
} else if (i == 4) {
var y_pred4 = y_pred
}
tf.concat([y_pred1, y_pred2, y_pred3, y_pred4])
此函数在 for 循环的每次迭代中存储 y_pred
的每个值。有没有一种方法可以提高效率,这样我就不需要在模型中包含大量的 if 语句。供您引用 y_pred
是一个张量。
仅供引用:i
代表迭代次数,因此如果达到迭代 1 y_pred1 = y_pred
最佳答案
如果每个张量都返回一个结果,您可能应该使用 Array
的实例。
如果张量的数量是静态的并且您事先知道:
var tf = new Array(number_of_tensors);
for (var i = 0; i < number_of_tensors; i++) {
var y_pred = /** get the result of your tensor **/
tf[i] = y_pred;
}
仅此而已。不需要concat()
,tf
已经是一个矩阵了。
如果您不知道张量的数量:
// before your for-loop
var tf = new Array();
// into your for-loop
var y_pred = /** get the result of your tensor **/
tf.push(y_pred);
关于javascript - 保存 Tensorflow.js 张量的先前值,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/63485453/