如何检查两个张量是否相等
How to check if two tensors are equal
给定任意秩的两个张量,我如何判断它们是否相同,我是否要设置我的正确解决方案,有任何类型的比较实现
要检查两个张量是否相等,可以使用tf.equal
。但它 returns 是一个张量,是按位运算的结果。这个张量的元素是1还是0。所以如果两个张量相等,计算后面的张量之和应该给出张量的元素个数。
const a = tf.tensor([1, 2, 3, 4], [2, 2]);
const b = tf.tensor([1, 2, 3, 4], [2, 2]);
const c = a.equal(b).sum().dataSync()[0]
console.log(c)
c === a.shape.reduce((a,b) => a *= b) ? console.log("true") : console.log("false")
<html>
<head>
<!-- Load TensorFlow.js -->
<script src="https://cdnjs.cloudflare.com/ajax/libs/tensorflow/0.12.4/tf.js"> </script>
</head>
<body>
</body>
</html>
给定任意秩的两个张量,我如何判断它们是否相同,我是否要设置我的正确解决方案,有任何类型的比较实现
要检查两个张量是否相等,可以使用tf.equal
。但它 returns 是一个张量,是按位运算的结果。这个张量的元素是1还是0。所以如果两个张量相等,计算后面的张量之和应该给出张量的元素个数。
const a = tf.tensor([1, 2, 3, 4], [2, 2]);
const b = tf.tensor([1, 2, 3, 4], [2, 2]);
const c = a.equal(b).sum().dataSync()[0]
console.log(c)
c === a.shape.reduce((a,b) => a *= b) ? console.log("true") : console.log("false")
<html>
<head>
<!-- Load TensorFlow.js -->
<script src="https://cdnjs.cloudflare.com/ajax/libs/tensorflow/0.12.4/tf.js"> </script>
</head>
<body>
</body>
</html>