more work

  always be 1. The output tensor will be of shape (6,6).

  >>> diff = cos_sim_same - tf.ones((6, 6), dtype=tf.dtypes.float32, name=None)

  >>> tf.reduce_sum(diff) == 0

  True

  >>> tf.reduce_sum(diff) == 0 # True

  <tf.Tensor: shape=(), dtype=bool, numpy=True>

  >>> cos_sim_same.shape

  TensorShape([6, 6])

  shape of the input tensors are both of shape `(256,512)`, the output tensor will

  be of shape `(256,256)`.

  >>> tf.reduce_sum(cos_sim_orth) == 0

  True

  >>> tf.reduce_sum(cos_sim_orth) == 0 # True

  <tf.Tensor: shape=(), dtype=bool, numpy=True> # True

  >>> cos_sim_orth.shape

  TensorShape([256, 256])