gpt4 book ai didi

javascript - 如何生成包含非零元素索引的张量?

转载 作者:搜寻专家 更新时间:2023-10-31 23:23:05 24 4
gpt4 key购买 nike

我有一个像

这样的张量
 [[0, 0, 0, 1, 0, 0],
[0, 3, 0, 0, 0, 1],
[0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0],
[0, 0, 0, 2, 0, 0],
[0, 0, 0, 0, 0, 0]]

我如何生成仅包含前者非零索引的张量?

[[0,3],[1,1],[1,5],[4,3]]

我在 tfjs 上,所以它似乎没有 boolean_mask 或其他人似乎在索引中使用的其他一些有用的功能。如果不使用 .data() 逃生口和对数组进行映射/过滤,这是否可能? (我只是将 tensorflow 用于其快速线性代数方法,所以这不会是最糟糕的,对吧?)

最佳答案

没错,目前没有可用于使用掩码定义分区的运算符。你可以考虑这个answer作为解决方法。但即使使用掩码,这里的问题也无法轻易解决,因为创建分区假设事先知道要检索的索引。

要解决这个问题,可以使用 logicalAnd在输入张量和所有元素都等于 1 的相同形状的张量之间。它很有用,因为如果初始元素为 0,它将返回与输入形状相同形状的张量,值为 0,否则为 1。使用 argMax , 可以检索等于 1 的元素的索引

let t = tf.tensor2d([[0, 0, 0, 1, 0, 0], [0, 3, 0, 0, 0, 1], [0, 0, 0, 0, 0, 0], [0, 0, 0, 0, 0, 0], [0, 0, 0, 2, 0, 0], [0, 0, 0, 0, 0, 0]])
let a = t.as1D()

const b = tf.cast(a, 'bool')
let helper = b.logicalAnd(tf.cast(tf.ones([a.size]), 'bool'))
const n = helper.sum().dataSync()[0]
const noNull = []
for(let i = 0; i < n ; i++) {
let ind = tf.argMax(helper)
let s = ind.dataSync()[0]
noNull.push(tf.argMax(helper).dataSync()[0])
if (s === 0) {
const [x, y] = helper.split([1, helper.size - 1])
helper = tf.concat(tf.tensor1d([0]), y)
} else if (s === helper.size) {
const [x, y] = helper.split([helper.size -1, 1])
helper = tf.concat(x, tf.tensor1d([0]))
} else {
const [x, _, y] = helper.split([s, 1, helper.size - s - 1])
helper = tf.concat([x,tf.tensor1d([0]), y])
}
}

const indexToCoords = (index, shape) => {
const pseudoShape = shape.map((a, b, c) => c.slice(b + 1).reduce((a, b) => a * b, 1))
let coords = []
let ind = index
for (let i = 0; i < shape.length; i++) {
coords.push(Math.floor(ind / pseudoShape[i]))
ind = ind % pseudoShape[i]
}
return coords
}

const coords = noNull.map(e => indexToCoords(e, t.shape))

console.log(coords)
<html>
<head>
<!-- Load TensorFlow.js -->
<script src="https://cdn.jsdelivr.net/npm/@tensorflow/tfjs@0.13.0"> </script>
</head>

<body>
</body>
</html>

关注此thread , 有一个更简单的方法来实现同样的事情

(async function() {
const x = tf.tensor2d(
[[0, 0, 0, 1, 0, 0],
[0, 3, 0, 0, 0, 1],
[0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0],
[0, 0, 0, 2, 0, 0],
[0, 0, 0, 0, 0, 0]]);

const mask = x.greater([0]).asType('bool');
const coords = await tf.whereAsync(mask);

coords.print();
}());
<html>
<head>
<!-- Load TensorFlow.js -->
<script src="https://cdn.jsdelivr.net/npm/@tensorflow/tfjs@0.13.0"> </script>
</head>

<body>
</body>
</html>

关于javascript - 如何生成包含非零元素索引的张量?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/52921033/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com