如何从 tensorflow 中的指定索引更新值

问题描述

我有两个张量,第一个 UIScene. 的形状为 A,其中 (B,N,2) 代表数字,N 代表索引。另一个 2一个形状 B(B,3) 是指定的值。我想将对应于 3B 值分散到形状 A。例如,

(B,H,W,3)

根据我的想法,形状为 B = [[[1 1] [0 2]] [[1 1] [1 2]]] A = [[[1 1 1] [2 0 1]] [[0 9 8] [9 1 2]]] 的最终张量是

(2,2,3)

除了 [[ [ [0,0],[0,[2,1] ],[ [0,[1,1,1],0] ] ],[ [ [0,0] ],9,8],[9,2] ] ]] 中指定的那些索引,其他位置将被设置为零。是否有解决此问题的解决方案?

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)