2

给定两个任意等级的张量，我如何判断它们是否相同，我是否要设置我的适当解决方案，是否有这种比较的任何类型的实现

4

1 回答 1

2

要检查两个张量是否相等，可以使用tf.equal. 但它返回一个张量，一个按位运算的结果。这个张量元素是 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>

于 2018-08-16T18:44:47.050 回答