问题描述
我有两个张量,第一个 UIScene.
的形状为 A
,其中 (B,N,2)
代表数字,N
代表索引。另一个 2
有一个形状 B
,(B,3)
是指定的值。我想将对应于 3
的 B
值分散到形状 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 (将#修改为@)