Skip to content

Commit 82a4cf1

Browse files
committed
update image_resize_short and shape doc
1 parent d1a8498 commit 82a4cf1

File tree

2 files changed

+7
-4
lines changed

2 files changed

+7
-4
lines changed

paddle/fluid/operators/shape_op.cc

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -36,10 +36,13 @@ class ShapeOpMaker : public framework::OpProtoAndCheckerMaker {
3636
public:
3737
void Make() override {
3838
AddInput("Input", "(Tensor), The input tensor.");
39-
AddOutput("Out", "(Tensor), The shape of input tensor.");
39+
AddOutput("Out",
40+
"(Tensor), The shape of input tensor, the data type of the shape"
41+
" is int64_t, will be on the same device with the input Tensor.");
4042
AddComment(R"DOC(
41-
Shape Operator.
42-
Get the shape of input tensor.
43+
Shape Operator
44+
45+
Get the shape of input tensor. Only support CPU input Tensor now.
4346
)DOC");
4447
}
4548
};

python/paddle/fluid/layers/nn.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4650,7 +4650,7 @@ def image_resize_short(input, out_short_len, resample='BILINEAR'):
46504650
46514651
Returns:
46524652
Variable: The output is a 4-D tensor of the shape
4653-
(num_batches, channls, out_h, out_w).
4653+
(num_batches, channls, out_h, out_w).
46544654
"""
46554655
in_shape = input.shape
46564656
if len(in_shape) != 4:

0 commit comments

Comments
 (0)