Method squeeze

ot/backend.py:3228–3229

(self, a, axis=None)

        ), f"Device discrepancy. First input is on {str(a_device)}, whereas second input is on {str(b_device)}"

    def squeeze(self, a, axis=None):
        return tnp.squeeze(a, axis=axis)

    def unsqueeze(self, a, axis):
        return tnp.expand_dims(a, axis=axis)