text.pad_along_dimension
Add padding to the beginning and end of data in a specific dimension.
text.pad_along_dimension(
data, axis=-1, left_pad=None, right_pad=None, name=None
)
Returns a tensor constructed from data
, where each row in dimension axis
is replaced by the concatenation of the left padding followed by the row
followed by the right padding. I.e., if L=left_pad.shape[0]
and
R=right_pad.shape[0]
, then:
result[i1...iaxis, 0:L] = left_pad
result[i1...iaxis, L:-R] = data[i0...iaxis]
result[i1...iaxis, -R:] = right_pad
Args |
data
|
<dtype>[O1...ON, A, I1...IM] A potentially ragged K dimensional
tensor with outer dimensions of size O1...ON ; axis dimension of size
A ; and inner dimensions of size I1...IM . I.e. K = N + 1 + M , where
N>=0 and M>=0 .
|
axis
|
An integer constant specifying the axis along which padding is added.
Negative axis values from -K to -1 are supported.
|
left_pad
|
<dtype>[L, I1...IM] An M+1 dimensional tensor that should be
prepended to each row along dimension axis ; or None if no padding
should be added to the left side.
|
right_pad
|
<dtype>[R, I1...IM] An M+1 dimensional tensor that should be
appended to each row along dimension axis ; or None if no padding
should be added to the right side.
|
name
|
The name of this op (optional).
|
Returns |
<dtype>[O1...ON, L + A + R, I1...IM]
A potentially ragged K dimensional tensor with outer dimensions of size
O1...ON ; padded axis dimension size L+A+R ; and inner dimensions of
size I1...IM . If data is a RaggedTensor , then the returned tensor
is a RaggedTensor with the same ragged_rank .
|
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.
Last updated 2024-12-20 UTC.
[null,null,["Last updated 2024-12-20 UTC."],[],[]]