Learning knot invariants across dimensions