关于将稀疏矩阵输入到图中

时间:2017-10-13 09:26:34

标签: tensorflow

由于数据维度太大,我必须将数据更改为稀疏矩阵,而不是密集数组。

但是,由于图形包含cnn,当我直接输入稀疏矩阵时,我被告知cnn无法接收稀疏张量。所以我必须先做“稀疏到密集”的操作。

但问题是我馈送的数据(多稀疏矩阵)应该转换为二维稀疏矩阵。(例如我有稀疏矩阵1,dim是[14,25500],稀疏矩阵2,dim是[ 14,25500],我想要的完美尺寸是[2,14,25500],但我面临的现实是[28,25500]) 所以我必须在输入图形后分割张量。

我想问一下,如果有其他办法可以解决这个问题吗?

1 个答案:

答案 0 :(得分:0)

tf.stack 是你的朋友

tf.stack([matrix1, matrix2]) # => [2,14,25500]