张量流中无法通过稠密层传递 3 阶稀疏张量

Question

假设我有这个稀疏张量：

import numpy as np
from tensorflow.sparse import SparseTensor, to_dense

indices = np.array( [[0,0,0]])
values = [1]
sparse_tensor = SparseTensor(indices = indices,values=values, dense_shape=(1,1,1))

和致密层

layer= Dense(5)

当试图通过致密层时

layer(sparse_tensor)

我收到错误消息：

尝试转换一个值(SparseTensor(indices=tf.Tensor([[0 0 0]], shape=(1, 3), dtype=int64), values=tf.Tensor([1.], shape=(1,), dtype=float32), dense_shape=tf.Tensor([1 1 1], 形状=(3,), dtype=int64))) 带有不支持的类型 () 到 a 张量。

然而，对 2 阶稀疏张量做同样的事情确实有效。

这是张量流的限制还是有解决方法？ （除了应用

to_dense

）。我正在使用 tensorflow 2.11.