[ Web Proxy ]
URL:
Viewing: https://www.tensorflow.org/api_docs/python/tf/sequence_mask [Back]  [Original]
tf.sequence_mask  |  TensorFlow v2.16.1 Skip to main content
TensorFlow [TensorFlow]
Install Learn API Ecosystem Community Why TensorFlow
Search:
GitHub
Overview Python C++ Java More
TensorFlow [TensorFlow]

tf.sequence_mask Stay organized with collections Save and categorize content based on your preferences.

Returns a mask tensor representing the first N positions of each cell.

View aliases

Compat aliases for migration

See Migration guide for more details.

tf.compat.v1.sequence_mask

tf.sequence_mask(
    lengths,
    maxlen=None,
    dtype=tf.dtypes.bool,
    name=None
)

Used in the notebooks

Used in the tutorials

If lengths has shape [d_1, d_2, ..., d_n] the resulting tensor mask has dtype dtype and shape [d_1, d_2, ..., d_n, maxlen], with

mask[i_1, i_2, ..., i_n, j] = (j < lengths[i_1, i_2, ..., i_n])

Examples:

tf.sequence_mask([1, 3, 2], 5)  # [[True, False, False, False, False],
                                #  [True, True, True, False, False],
                                #  [True, True, False, False, False]]

tf.sequence_mask([[1, 3],[2,0]])  # [[[True, False, False],
                                  #   [True, True, True]],
                                  #  [[True, True, False],
                                  #   [False, False, False]]]

lengths integer tensor, all its values <= maxlen.
maxlen scalar integer tensor, size of last dimension of returned tensor. Default is the maximum value in lengths.
dtype output type of the resulting tensor.
name name of the op.

A mask tensor of shape lengths.shape + (maxlen,), cast to specified dtype.

ValueError if maxlen is not a scalar.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.

Last updated 2024-04-26 UTC.

[[["Easy to understand","easyToUnderstand","thumb-up"],["Solved my problem","solvedMyProblem","thumb-up"],["Other","otherUp","thumb-up"]],[["Missing the information I need","missingTheInformationINeed","thumb-down"],["Too complicated / too many steps","tooComplicatedTooManySteps","thumb-down"],["Out of date","outOfDate","thumb-down"],["Samples / code issue","samplesCodeIssue","thumb-down"],["Other","otherDown","thumb-down"]],["Last updated 2024-04-26 UTC."],[],[]]
Web Proxy Viewer  |  New URL  |  Original Page