当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python tf.strings.unsorted_segment_join用法及代码示例


基于 segment_ids 加入 inputs 的元素。

用法

tf.strings.unsorted_segment_join(
    inputs, segment_ids, num_segments, separator='', name=None
)

参数

  • inputs Tensor 类型为 string 。要加入的输入。
  • segment_ids 一个Tensor。必须是以下类型之一:int32 , int64。一个张量,其形状是 data.shape 的前缀。不支持负分段 ID。
  • num_segments 一个Tensor。必须是以下类型之一:int32 , int64。一个标量。
  • separator 可选的 string 。默认为 "" 。加入时使用的分隔符。
  • name 操作的名称(可选)。

返回

  • Tensor 类型为 string

计算沿张量段的字符串连接。给定 segment_ids 排名为 Ndata 排名为 N+M

`output[i, k1...kM] = strings.join([data[j1...jN, k1...kM])`

连接在所有 [j1...jN] 上,使得 segment_ids[j1...jN] = i。字符串以行优先顺序连接。

例如:

inputs = [['Y', 'q', 'c'], ['Y', '6', '6'], ['p', 'G', 'a']]
output_array = string_ops.unsorted_segment_join(inputs=inputs,
                                                segment_ids=[1, 0, 1],
                                                num_segments=2,
                                                separator=':'))
# output_array ==> [['Y', '6', '6'], ['Y:p', 'q:G', 'c:a']]


inputs = ['this', 'is', 'a', 'test']
output_array = string_ops.unsorted_segment_join(inputs=inputs,
                                                segment_ids=[0, 0, 0, 0],
                                                num_segments=1,
                                                separator=':'))
# output_array ==> ['this:is:a:test']

相关用法


注:本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.strings.unsorted_segment_join。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。