老师,想问一下关于gram_matrix的问题
来源:6-8 图像风格转换计算图构建与损失函数计算
好好过生活
2018-12-10
在gram_matrix函数那一部分,求gram矩阵时,features是个3维的(我跑的代码维度(1, 784, 512))),为什么第一个维度可以不管?直接进行转置相乘
写回答
1回答
-
同学你好,可以参考一下官方的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.
我理解,它的意思就是最后的两个维度的矩阵进行矩阵乘法。
012018-12-11
相似问题