在TF 2.1中展平一个非常参差不齐的张量

时间:2020-05-07 13:58:57

标签: python tensorflow reshape

我终于迁移到TF 2.1,并且尝试将代码移植到本机TF 2.1。所以这是一个问题:如何将各种等级的衣衫agged的张量整形为一维张量?本质上,我有一组权重矩阵,例如。 nodes_in_community9 = [node for node,community in partition.items() if community == 9] S = G.subgraph(nodes_in_community9) (这些是形状,而不是值),我想将整个怪物扁平化为一维张量。最有效的方法是什么?更好的是,我该如何写一个函数来展平任意长度的任意程度的张量的形状参差?

当前,我正在使用以下解决方案,但我怀疑Python列表可能会大大降低所有速度:

[3,3,1,32], [32], [21632,20], [20], [20,10], [10]

0 个答案:

没有答案