TensorFlow函數(shù)：tf.substr

tf.substr函數(shù)

tf.substr(
    input,
    pos,
    len,
    name=None
)

參見指南：字符串操作>拆分

從字符串的 Tensor 中返回子字符串.

對于輸入 Tensor 中的每個字符串,創(chuàng)建一個從索引 pos 開始的子字符串,總長度為 len.

如果 len 定義了一個將超出輸入字符串長度的子字符串,那么使用盡可能多的字符.

如果 pos 是負數(shù)或指定的字符索引大于任何輸入字符串,則會拋出一個 InvalidArgumentError.

pos 和 len 必須具有相同的形狀,否則在 Op 創(chuàng)建時會拋出一個 ValueError.

注意：Substr 支持最多兩個維度的廣播.

例子

使用標(biāo)量 pos 和 len：

input = [b'Hello', b'World']
position = 1
length = 3

output = [b'ell', b'orl']

使用與 input 具有相同形狀的 pos 和 len：

input = [[b'ten', b'eleven', b'twelve'],
         [b'thirteen', b'fourteen', b'fifteen'],
         [b'sixteen', b'seventeen', b'eighteen']]
position = [[1, 2, 3],
            [1, 2, 3],
            [1, 2, 3]]
length =   [[2, 3, 4],
            [4, 3, 2],
            [5, 5, 5]]

output = [[b'en', b'eve', b'lve'],
          [b'hirt', b'urt', b'te'],
          [b'ixtee', b'vente', b'hteen']]

將 pos 和 len 廣播到 input：

input = [[b'ten', b'eleven', b'twelve'],
         [b'thirteen', b'fourteen', b'fifteen'],
         [b'sixteen', b'seventeen', b'eighteen'],
         [b'nineteen', b'twenty', b'twentyone']]
position = [1, 2, 3]
length =   [1, 2, 3]

output = [[b'e', b'ev', b'lve'],
          [b'h', b'ur', b'tee'],
          [b'i', b've', b'hte'],
          [b'i', b'en', b'nty']]

將 input 廣播到 pos 和 len：

input = b'thirteen'
position = [1, 5, 7]
length =   [3, 2, 1]

output = [b'hir', b'ee', b'n']

函數(shù)參數(shù)：

• input：一個 string 類型的 Tensor,strings 張量.
• pos：一個 Tensor,必須是以下類型之一：int32,int64,標(biāo)量定義每個子字符串中第一個字符的位置.
• len：一個 Tensor,必須具有相同的類型pos,標(biāo)量定義要包含在每個子字符串中的字符數(shù).
• name：操作的名稱(可選).

函數(shù)返回值：

函數(shù)返回一個 string 類型的 Tensor.