-
Notifications
You must be signed in to change notification settings - Fork 5.6k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Support uint8 for fill_constant #31911
Conversation
Thanks for your contribution! |
✅ This PR's description meets the template requirements! |
920925b
to
929925f
Compare
929925f
to
81d6694
Compare
81d6694
to
8f947cb
Compare
b5d3424
to
3bc1d5b
Compare
9fa5b2a
to
646f5f7
Compare
646f5f7
to
a031ef7
Compare
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
a031ef7
to
6e01aa6
Compare
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
@@ -635,7 +635,7 @@ def fill_constant(shape, dtype, value, force_cpu=False, out=None, name=None): | |||
If ``shape`` is a list or tuple, the elements of it should be integers or Tensors with shape [1]. | |||
If ``shape`` is an Tensor, it should be an 1-D Tensor with date type int32 or int64. | |||
dtype(np.dtype|str): Data type of the output Tensor which can | |||
be float16, float32, float64, int32, int64. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
python/paddle/tensor/creation.py里面2.0的接口,ones
、zeros
、full
的文档里面也加一下uint8,可以下个PR再加。
PR types
New features
PR changes
APIs
Describe
Support uint8 for fill_constant