“6460985d8d0068da6402eb59aa2cee1d597b3612”上不存在“ppstructure/git@gitcode.net:s920243400/PaddleOCR.git”
未验证 提交 46a6950e 编写于 作者: Z zxcd 提交者: GitHub

add dtype param for arange API. (#10075)

上级 a46a0610
...@@ -46,7 +46,7 @@ def positionalencoding2d(d_model, height, width): ...@@ -46,7 +46,7 @@ def positionalencoding2d(d_model, height, width):
# Each dimension use half of d_model # Each dimension use half of d_model
d_model = int(d_model / 2) d_model = int(d_model / 2)
div_term = paddle.exp( div_term = paddle.exp(
paddle.arange(0., d_model, 2) * -(math.log(10000.0) / d_model)) paddle.arange(0., d_model, 2, dtype='int64') * -(math.log(10000.0) / d_model))
pos_w = paddle.arange(0., width, dtype='float32').unsqueeze(1) pos_w = paddle.arange(0., width, dtype='float32').unsqueeze(1)
pos_h = paddle.arange(0., height, dtype='float32').unsqueeze(1) pos_h = paddle.arange(0., height, dtype='float32').unsqueeze(1)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册