How to perform tensor_scatter_nd_update on batch of data

Question

I am trying to modify some elements from a tensorflow tensor. The desired operation looks like this:

input=[[1,2,3],[4,5,6],[7,8,9],...]
update=[-40,-50,-60,...]