TensorFlow函数:tf.sparse_slice

2018-03-09 10:38 更新

tf.sparse_slice 函数

sparse_slice(
    sp_input,
    start,
    size,
    name=None
)

定义在:tensorflow/python/ops/sparse_ops.py.

在 tf.sparse_slice 函数中根据 start 和 size 对 SparseTensor 进行切片.

例如,如果输入是为以下的内容:

input_tensor = shape = [2, 7]
[    a   d e  ]
[b c          ]

以图形方式输出张量为:

sparse_slice([0, 0], [2, 4]) = shape = [2, 4]
[    a  ]
[b c    ]

sparse_slice([0, 4], [2, 3]) = shape = [2, 3]
[ d e  ]
[      ]

函数参数:

  • sp_input:表示要分割的 SparseTensor.
  • start:1-D,张量代表分割的开始.
  • size:1-D,张量表示分割的大小.
  • name:操作的名称(可选).

函数返回值:

由拼接产生的 SparseTensor 对象.

可能引发的异常:

  • TypeError:如果 sp_input 不是 SparseTensor,则会引发此异常.
以上内容是否对您有帮助:
在线笔记
App下载
App下载

扫描二维码

下载编程狮App

公众号
微信公众号

编程狮公众号