提交 e751d4f3 authored 作者: Frederic's avatar Frederic

fix typo that cause crash.

上级 078420b8
...@@ -1924,7 +1924,7 @@ class GpuAdvancedSubtensor1(tensor.AdvancedSubtensor1, GpuOp): ...@@ -1924,7 +1924,7 @@ class GpuAdvancedSubtensor1(tensor.AdvancedSubtensor1, GpuOp):
assert self.perform_using_take == True, ( assert self.perform_using_take == True, (
"GpuAdvancedSubtensor1 used the fast version") "GpuAdvancedSubtensor1 used the fast version")
if idx.dtype != numpy.int64: if idx.dtype != numpy.int64:
if idx.dtype in [numpy.int8, numpyt.int16, numpy.int32, if idx.dtype in [numpy.int8, numpy.int16, numpy.int32,
numpy.int64, numpy.uint8, numpy.uint16, numpy.int64, numpy.uint8, numpy.uint16,
numpy.uint32]: numpy.uint32]:
idx = idx.astype(numpy.int64) idx = idx.astype(numpy.int64)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论