javascript - 按列对矩阵进行排序

标签 javascript tensorflow matrix tensorflow.js

我有一个看起来像这样的矩阵:

[[5,2],
[4,3],
[3,4]]
使用命令
tf.tensor([...])
我想按索引为 0 的列排列矩阵,这样它看起来像这样:
[[3,4],
[4,3],
[5,2]]
我将如何使用 Tensorflow.js 做到这一点?

最佳答案

看这个例子,张量在第一个轴上排序。
目前还没有tf.sort在 tensorflow.js 中。但是为了达到同样的效果,我们可以在第一个轴上对张量进行切片,然后对其进行排序并获取索引。使用后者,我们可以使用 tf.gather 对初始张量进行排序

a = tf.tensor2d([[3,4], [4,3],[5,2]])
firstAxis = a.gather([0], 1).reshape([-1]);
ind = tf.topk(firstAxis, a.shape[0]).indices
a.gather(ind.reverse(), 0).print()

关于javascript - 按列对矩阵进行排序,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/66718587/

相关文章:

javascript - 使用Java脚本验证PHP表单数据

javascript - Node.js 服务 HTML,执行命令行和回调函数

python - TensorFlow:按名称访问变量的内容

c# - 矩阵/坐标转换顺序

javascript - 如何将绑定(bind)到 KoGrid 单元格的项目传递给 ViewModel

javascript - 结合Lodash方法

python - 在 Anaconda 上安装特定版本的 TensorFlow

c# - 当我将图像提供给经过训练的模型时,它会为未在模型中训练的对象提供 80% 以上的准确度

python - 将Numpy数组中的所有行对相乘

java - 矩阵上的递归运动