老师,想问一下关于gram_matrix的问题

来源:6-8 图像风格转换计算图构建与损失函数计算

好好过生活

2018-12-10

在gram_matrix函数那一部分,求gram矩阵时,features是个3维的(我跑的代码维度(1, 784, 512))),为什么第一个维度可以不管?直接进行转置相乘

写回答

1回答

正十七

2018-12-11

同学你好,可以参考一下官方的api文档: https://www.tensorflow.org/api_docs/python/tf/linalg/matmul

里面有这么一段:The inputs must, following any transpositions, be tensors of rank >= 2 where the inner 2 dimensions specify valid matrix multiplication arguments, and any further outer dimensions match.

我理解,它的意思就是最后的两个维度的矩阵进行矩阵乘法。

0
1
好好过生活
非常感谢老师!
2018-12-11
共1条回复

深度学习之神经网络(CNN/RNN/GAN)算法原理+实战

深度学习算法工程师必学,深入理解深度学习核心算法CNN RNN GAN

2617 学习 · 935 问题

查看课程