System and machine learning method for localization of an input device relative to a touch sensitive surface

A method comprising:

• at an electronic device in communication with a touch sensitive surface:
• applying, via drive circuitry coupled to the touch sensitive surface, drive signals to one or more first touch electrodes of the touch sensitive surface;
• sensing, via sense circuitry coupled to the touch sensitive surface, sense signals at one or more second touch electrodes of the touch sensitive surface;
• estimating, via a machine learning algorithm, a location of a stylus based on at least a subset of the sense signals from a first subset of the second touch electrodes;
• and
• estimating, via a second machine learning algorithm, a distance of the stylus from the touch sensitive surface based on at least a second subset of the sense signals sensed from a second subset of the second touch electrodes different from the first subset of the second touch electrodes.

A non-transitory computer readable storage medium storing instructions that, when executed by an electronic device with one or more processors and in communication with a touch sensitive surface cause the electronic device to perform a method comprising:

• applying, via drive circuitry coupled to the touch sensitive surface, drive signals to one or more first touch electrodes of the touch sensitive surface;
• sensing, via sense circuitry coupled to the touch sensitive surface, sense signals at one or more second touch electrodes of the touch sensitive surface;
• estimating, via a machine learning algorithm, a location of a stylus based on at least a subset of the sense signals from a first subset of the second touch electrodes;
• and
• estimating, via a second machine learning algorithm, a distance of the stylus from the touch sensitive surface based on at least a second subset of the sense signals sensed from a second subset of the second touch electrodes different from the first subset of the second touch electrodes.