search

tensorflow / community

Stores documents used by the TensorFlow developer community
Apache License 2.0
1.26k stars 576 forks source link

RFC: Tensor Indexed Updates #433

Open cantonios opened 1 year ago

cantonios commented 1 year ago

This is to allow numpy-like (or rather JAX-like) indexed tensor modifications like slice assignment.

fchollet commented 1 year ago

This is great, thanks for the proposal. LGTM.

Note that this is currently doable in TF in a performant way via a XLA op:

from tensorflow.compiler.tf2xla.python import xla

output = xla.dynamic_update_slice(input, update, indices)
tanzhenyu commented 1 year ago

This is a great initiative!

dimentary commented 1 year ago

Thanks for the proposal!

Ayazzia01 commented 1 year ago

@theadactyl @ematejska

Please.