代码之家  ›  专栏  ›  技术社区  ›  Avan

如何得到稀疏张量中使用的张量的名称?

  •  0
  • Avan  · 技术社区  · 4 年前

    TensorFlow将稀疏张量表示为三个独立的稠密张量:索引、值和稠密形状。在Python中,为了便于使用,这三个张量被收集到一个SparseTensor类中。如果有单独的索引、值和密集的u形张量,请在传递到下面的ops之前将它们包装在SparseTensor对象中。

    我的问题是,给定一个稀疏张量或稀疏张量值,如何检索每个组成张量的单独名称,即索引张量、值张量和形状张量?(除非我碰巧误解了引用文本中的解释,这是不可能的?)

    1 回复  |  直到 4 年前
        1
  •  1
  •   Armadillan    4 年前

    您可以使用以下方法访问SparseTensor对象中的三个张量中的每一个:

    # Here some_sparse_tensor is an object of the tf.sparse.SparseTensor class
    some_sparse_tensor.indices
    some_sparse_tensor.values
    some_sparse_tensor.dense_shape
    

    根据文件: https://www.tensorflow.org/api_docs/python/tf/sparse/SparseTensor#attributes

    您无法获取它们的“名称”,但可以将它们分配给它们自己的变量: values_of_sparse_tensor = some_sparse_tensor.values