我想对形状为 [a,b,c] 的张量中的元素进行倍增，每个元素在第 3 维中都有一个来自形状为 [a,b] 的张量的标量。
例如，
x =
|[1,2][3,4][5,6]|
|[1,2][3,4][5,6]|
|[1,2][3,4][5,6]|
y =
|0 0 1|
|1 0 0|
|0 1 0|
我想乘以 x 和 y 以获得：
res =
|[0,0][0,0][5,6]|
|[1,2][0,0][0,0]|
|[0,0][3,4][0,0]|
你能告诉我任何方法来实现这一目标吗？
PS：也许我可以做一个y'由y它
y' =
|[0,0][0,0][1,1]|
|[1,1][0,0][0,0]|
|[0,0][1,1][0,0]|
然后我可以tf.mul()用来获得结果。但是我找不到任何以这种方式复制张量中的元素的东西。
tensorflow ×1