community icon indicating copy to clipboard operation
community copied to clipboard

Published 20 hours ago • verified publisher icon tensorflow

RFC: Tensor Indexed Updates

Open cantonios opened this issue 2 years ago • 4 comments

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

cantonios avatar Oct 24 '22 19:10 cantonios

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)

fchollet avatar Dec 01 '22 18:12 fchollet

This is a great initiative!

tanzhenyu avatar Dec 01 '22 18:12 tanzhenyu

Thanks for the proposal!

dimentary avatar Dec 06 '22 11:12 dimentary

@theadactyl @ematejska

Please.

Ayazzia01 avatar Aug 14 '23 18:08 Ayazzia01