forked from PaddlePaddle/Paddle
-
Notifications
You must be signed in to change notification settings - Fork 0
/
unzip.py
85 lines (74 loc) · 2.78 KB
/
unzip.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
# Copyright (c) 2023 PaddlePaddle Authors. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
from paddle.base.data_feeder import check_variable_and_dtype
from paddle.base.layer_helper import LayerHelper
def unzip(input, lod):
r"""
**unzip layers**
unzip 'input' accroding to 'lod'
Args:
input (Variable): The zipped input, 2-D LodTensor with shape [N, M].
lod (Variable): The original lod of unzipped input, 1-D LodTensor with shape[K].
Returns:
Variable: The original unzipped tensor, 2-D LodTensor with shape[K-1, M].
Examples:
.. code-block:: python
>>> # doctest: +REQUIRES(env:GPU)
>>> import numpy as np
>>> import paddle
>>> paddle.set_device('gpu')
>>> input_np = np.array([
... [1.0, 2.0, 3.0, 4.0],
... [10.0, 20.0, 30.0, 40.0],
... [100.0, 200.0, 300.0, 400.0]
... ])
>>> lod_np = np.array([0, 4, 4, 8, 8, 8, 8, 12, 12, 12, 12])
>>> input = paddle.to_tensor(input_np, "int64")
>>> lod = paddle.to_tensor(lod_np, "int64")
>>> unzipped_input = paddle.incubate.operators.unzip(input, lod)
>>> print(unzipped_input)
Tensor(shape=[10, 4], dtype=int64, place=Place(gpu:0), stop_gradient=True,
[[1 , 2 , 3 , 4 ],
[0 , 0 , 0 , 0 ],
[10, 20, 30, 40],
[0 , 0 , 0 , 0 ],
[0 , 0 , 0 , 0 ],
[0 , 0 , 0 , 0 ],
[100, 200, 300, 400],
[0 , 0 , 0 , 0 ],
[0 , 0 , 0 , 0 ],
[0 , 0 , 0 , 0 ]])
"""
helper = LayerHelper('unzip', **locals())
out = helper.create_variable(dtype=input.dtype)
check_variable_and_dtype(
input,
'input',
[
'float16',
'float32',
'float64',
'int',
'bool',
'int64',
'complex64',
'complex128',
],
'unzip',
)
check_variable_and_dtype(lod, 'lod', ['int', 'int64'], 'unzip')
helper.append_op(
type='unzip', inputs={'X': [input], 'lod': [lod]}, outputs={'Y': [out]}
)
return out