TensorDot与轴心缩减

Question

我有两个张量a和b，像[?,200,200,48]，我想在轴4上得到点积，得到一个类似[?,200,200,1]的张量。

我尝试使用tf.tensordot(a,b,axis=[[3][3]])，结果是张量类似于[?,200,200,?,200,200]

我在哪里不好？

Answer 1

试试这个：

from tensorflow.keras.layers import *
import tensorflow.keras.backend as K
import tensorflow as tf

BS = 100
x = K.random_normal((BS, 200, 200, 48))
y = K.random_normal((BS, 200, 200, 48))

K.sum(tf.multiply(x, y[:tf.newaxis]), axis=-1, keepdims=True).shape

详细说明：https://stackoverflow.com/questions/51657476/keras-dot-dot-layer-behavior-on-3d-tensors