如何创建混合稀疏张量?

时间:2019-09-29 05:08:43

标签: tensorflow

我要将模型从pytorch移植到tensorflow。 Pytorch的张量类型称为混合稀疏,这意味着只有前几个维度是稀疏的,最后一个维度可以是密集的。张量流中有类似的东西吗?我已经检查了tf.sparse.SparseTensor api,它不支持混合张量。我该如何解决?我正在知识图上进行关系嵌入和节点嵌入,因此我需要使用混合稀疏张量。

我正在考虑将混合稀疏张量分成多个“正常”稀疏张量,进行一些计算并将结果重新组合在一起。我想知道这是否是个好方法?

0 个答案:

没有答案