在tensorflow的早期版本中, axes=0 会引发 ValueError: 'axes' must be at least 1. . 某种程度上 tf.tensordot() 曾经需要至少一个维度来实际总结 . 简单的方法是简单地使用 tf.expand_dims() 添加"fake"维度 .
在 tensorflow<=1.5 上,您可以通过以下方式获得与上述相同的结果:
M = tf.tensordot(tf.expand_dims(A, 0), tf.expand_dims(B, 0), axes=[[0],[0]])
a_np = np.array([[1, 2, 3]]) # shape: (1,3) [a row vector], 2D Tensor
b_np = np.array([[4], [5], [6]]) # shape: (3,1) [a column vector], 2D Tensor
a = tf.placeholder(dtype='float32', shape=[1, 3])
b = tf.placeholder(dtype='float32', shape=[3, 1])
c = a*b # Result: an outer-product of a,b
d = tf.multiply(a,b) # Result: an outer-product of a,b
e = tf.tensordot(a,b, axes=[0,1]) # Result: an outer-product of a,b
With a batch:
a_np = np.array([[[1, 2, 3]], [[4, 5, 6]]]) # shape: (2,1,3) [a batch of two row vectors], 3D Tensor
b_np = np.array([[[7], [8], [9]], [[10], [11], [12]]]) # shape: (2,3,1) [a batch of two column vectors], 3D Tensor
a = tf.placeholder(dtype='float32', shape=[None, 1, 3])
b = tf.placeholder(dtype='float32', shape=[None, 3, 1])
c = a*b # Result: an outer-product per batch
d = tf.multiply(a,b) # Result: an outer-product per batch
e = tf.tensordot(a,b, axes=[1,2]) # Does NOT result with an outer-product per batch
6 回答
是的,您可以通过利用tensorflow的广播语义来实现这一点 . 将第一个输出调整为自身的大小1xN,将第二个调整为自身的大小Mx1,并且当您将它们相乘时,您将获得所有结果的MxN广播 .
(您可以在numpy中使用相同的东西来查看它在更简单的上下文中的行为,顺便说一句:
在tensorflow中你究竟是如何做到这一点取决于你想要使用哪些轴以及你想要的结果乘法的语义,但一般的想法适用 .
如果其他人偶然发现这一点,根据tensorflow文档,您可以使用tf.einsum()函数来计算两个张量
a
和b
的外积:有点令人惊讶的是,直到最近,在张量流中,在任意张量(也称为“张量积”)之间进行外部产品没有简单而“自然”的方式,特别是考虑到图书馆的名称......
使用
tensorflow>=1.6
,你现在可以通过简单的方式获得你想要的东西:在tensorflow的早期版本中,
axes=0
会引发ValueError: 'axes' must be at least 1.
. 某种程度上tf.tensordot()
曾经需要至少一个维度来实际总结 . 简单的方法是简单地使用tf.expand_dims()
添加"fake"维度 .在
tensorflow<=1.5
上,您可以通过以下方式获得与上述相同的结果:这为两个张量的位置0添加了维度1的新索引,然后让
tf.tensordot()
对这些索引求和 .无论是否使用批处理,tf.multiply(及其'*'快捷方式)都会生成外部产品 . 特别是,如果两个输入张量具有[batch,n,1]和[batch,1,n]的3D形状,则此op将计算每个[n,1],[1,n]的外积 . 批次中的样品 . 如果没有批处理,那么两个输入张量是2D,这个op将计算外部产品相同 . 另一方面,虽然tf.tensordot产生2D矩阵的外部产品,但是当添加批次时它没有类似地广播 .
Without a batch:
With a batch:
Running any of these two graphs:
正如其他答案中所指出的,外部产品可以使用广播来完成:
说明:
a[..., None]
在最后一个轴之后插入长度为1的新尺寸 .类似地,
b[None, ...]
在第一个轴之前插入长度为1的新维度 .元素范围的乘法然后从形状
(10, 1) * (1, 5)
到(10, 5) * (10, 5)
广播张量,计算外部产品 .插入的位置附加尺寸确定计算外部产品的尺寸 . 例如,如果两个张量都具有批量大小,则可以使用
:
跳过它,这将给出a[:, ..., None] * b[:, None, ...]
. 这可以进一步缩写为a[..., None] * b[:, None]
. 要在最后一个维度上执行外部产品并因此支持任意数量的批次维度,请使用a[..., None] * b[..., None, :]
.我会评论MasDra,但不会让我作为新的注册用户 .
可以通过以下方式获得排列在长度顺序列表U中的多个向量的一般外积