Tensorflow 将张量值字典分配给更大的张量

我使用的是Tensorflow 2.x，我有一个深度字典

depth\u dict

：它以[N，d]的形式包含大小为d的N个不同元素。 我也有大小为[H，W，1]的

索引

张量，包含从0到N-1到

深度dict

中H，W网格每个元素的位置。我的想法是更新大小为[H，W，4]的

深度图

（与

索引

相同的前两个维度，与

深度图

中的元素大小相同的第三个维度），使用可由

深度图

中的

索引

张量定位的值，即

深度图[i，j，：]=深度图[i，j]，：]，但是在全张量格式上。如果不显式地使用for循环，我可以使用什么函数或想法