Unverified 提交 5a6ca39c authored 作者: abergeron's avatar abergeron 提交者: GitHub

Merge pull request #6665 from justanhduc/master

fixed bilinear_kernel_1D to return a float32 tensor
...@@ -1766,7 +1766,7 @@ def bilinear_kernel_1D(ratio, normalize=True): ...@@ -1766,7 +1766,7 @@ def bilinear_kernel_1D(ratio, normalize=True):
kern = T.concatenate([half_kern, half_kern[-2::-1]]) kern = T.concatenate([half_kern, half_kern[-2::-1]])
if normalize: if normalize:
kern /= ratio kern /= T.cast(ratio, theano.config.floatX)
return kern return kern
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论