javascript - 保存 Tensorflow.js 张量的先前值
问题描述
我有一个函数可以存储张量的所有值,然后将其连接到一个矩阵中。该函数如下所示:
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])
此函数存储y_pred
for 循环的每次迭代中的每个值。有没有办法让这更有效,这样我就不需要在模型内部有大量的 if 语句。供您参考y_pred
的是张量。
仅供参考:i
代表迭代次数,因此如果达到迭代 1y_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);
推荐阅读
- asp.net - 如何在 Asp.net 中隐藏错误请求 400 的响应标头
- python-3.x - 格式化电脑后出现编译问题
- php - 如何在 CodeIgniter 中使用 JWPlayer?
- docusaurus - 使侧边栏不可折叠并始终在 docusaurus V2 经典预设中展开
- python - Visual Studio Code:在没有相应 .py 文件的情况下使用 .pyd 文件时可以使用 Intellisense /“转到定义”吗?(pywin32)
- c# - 如何在 Polly 中创建策略以支持 http 响应 retry-after 标头
- vue.js - vue渲染的问题 Json返回良好但不显示
- java - 如何分别打印给定数字的每个数字?
- unity3d - Forge AR/VR Toolkit 无法在 Unity 中获取场景
- mongodb - 具有 $text 搜索的 MongoDB $match 块中的顺序是否重要?