[RFC] Tensor Indexed Updates

[Antonio Sanchez]

Hello everyone,

I am happy to announce that RFC: Tensor Indexed Updates is ready for review.

The feedback phase will be open until November 11, 2022. Comments are invited to the pull request linked. You can view the design doc here, and also leave comments inline on the document source.

Summary:

This proposal introduces a new python API for "updating" elements within a tensor using numpy-like indexing.

• For immutable tensors, introduces tensor.at[indices].set(updates) to return a modified copy of the original tensor (follows the JAX syntax)
• For tf.Variable, allows direct slice assignment, var[indices] = updates.

Thanks for your review and participation!

~Antonio