what is torch's unsqueeze equivalence with tensorflow?
what is torch's unsqueeze  equivalence with tensorflow? 

#tensorflow auto-broadcasts singleton dimensions
lower_bounds = tf.argmax(set_1[:, :2].unsqueeze(1), set_2[:, :2].unsqueeze(0))  # (n1, n2, 2)
upper_bounds = tf.argmin(set_1[:, 2:].unsqueeze(1), set_2[:, 2:].unsqueeze(0))  # (n1, n2, 2)


Maybe you wanna try this:
tf.expand_dims(x, axis)
