refine unsqueeze, test=develop #25470
Merged
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
|
Thanks for your contribution! |
zhiqiu
force-pushed
the
dev/refine_unsqueeze
branch
from
49b4466
to
66d1a0a
Compare
lanxianghit
reviewed
Jul 28, 2020
python/paddle/tensor/manipulation.py
Outdated
| @@ -556,7 +577,7 @@ def squeeze(x, axis=None, name=None): | |||
| name (str, optional): Please refer to :ref:`api_guide_Name`, Default None. | |||
|
|
|||
| Returns: | |||
| Tensor: Output squeezed Tensor. Data type is same as input Tensor. | |||
| Tensor: Output squeezed Tensor with the same data type as input Tensor. | |||
python/paddle/tensor/manipulation.py
Outdated
| input (Variable): The input Tensor to be unsqueezed. It is a N-D Tensor of data types float32, float64, int32. | ||
| axes (int|list|tuple|Variable): Indicates the dimensions to be inserted. The data type is ``int32`` . If ``axes`` is a list or tuple, the elements of it should be integers or Tensors with shape [1]. If ``axes`` is an Variable, it should be an 1-D Tensor . | ||
| name (str|None): Name for this layer. | ||
| x (Tensor): The input Tensor to be unsqueezed. It is a N-D Tensor of data types float32, float64, int32. |
There was a problem hiding this comment.
不支持int64？bool类型呢，是否需要支持？
There was a problem hiding this comment.
int64是支持的，bool在c++ kernel中未支持。
python/paddle/tensor/manipulation.py
Outdated
|
|
||
| Returns: | ||
| Variable: Output unsqueezed Tensor, with data type being float32, float64, int32, int64. | ||
| Tensor: Output unsqueezed Tensor with the same data type as input Tensor. |
|
|
||
| out2 = paddle.unsqueeze(x, axis=[0, 2]) | ||
| print(out2.shape) # [1, 5, 1, 10] | ||
|
|
|
|
||
| return out | ||
| return layers.unsqueeze(x, axis, name) |
There was a problem hiding this comment.
直接复用了layers.squeeze的报错，报错中 axes -> axis/axes。
zhiqiu
force-pushed
the
dev/refine_unsqueeze
branch
from
7e246e5
to
064618a
Compare
|
|
lanxianghit
reviewed
Aug 7, 2020
python/paddle/fluid/layers/nn.py
Outdated
| @@ -6253,12 +6254,12 @@ def unsqueeze(input, axes, name=None): | |||
| then Unsqueezed tensor with axes=[0, 4] has shape [1, 3, 4, 5, 1]. | |||
|
|
|||
| Args: | |||
| input (Variable): The input Tensor to be unsqueezed. It is a N-D Tensor of data types float32, float64, int32. | |||
| input (Variable): The input Tensor to be unsqueezed. Support data type: float32, float64, bool, int8, int32, int64. | |||
There was a problem hiding this comment.
Suggested change
| input (Variable): The input Tensor to be unsqueezed. Support data type: float32, float64, bool, int8, int32, int64. | |
| input (Variable): The input Tensor to be unsqueezed. Supported data type: float32, float64, bool, int8, int32, int64. |
python/paddle/tensor/manipulation.py
Outdated
| @@ -473,18 +494,19 @@ def squeeze(x, axis=None, name=None): | |||
| out.shape = [1, 3, 5] | |||
|
|
|||
| Args: | |||
| input (Tensor): The input Tensor. Support data type: float32, float64, int8, int32, int64. | |||
| x (Tensor): The input Tensor. Support data type: float32, float64, bool, int8, int32, int64. | |||
There was a problem hiding this comment.
Suggested change
| x (Tensor): The input Tensor. Support data type: float32, float64, bool, int8, int32, int64. | |
| x (Tensor): The input Tensor. Supported data type: float32, float64, bool, int8, int32, int64. |
zhiqiu
force-pushed
the
dev/refine_unsqueeze
branch
from
fccb560
to
6860851
Compare
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
PR types
Breaking changes
PR changes
APIs
Describe
Refine paddle.unsqueeze
axissupportsint, list, tupledtypesupports float32, float64, bool, int8, int32, int64, (and float16 on CUDAPlace).