提交 c007e003 authored 作者: Frederic Bastien's avatar Frederic Bastien

allow int16 in subtensor.

上级 214e9026
......@@ -2605,7 +2605,7 @@ class Subtensor(Op):
@staticmethod
def convert(entry, slice_ok=True):
scal_types = [scal.int64, scal.int32, scal.int16, scal.int8]
tensor_types = [bscalar, iscalar, lscalar]
tensor_types = [lscalar, iscalar, wscalar, bscalar]
if isinstance(entry, gof.Variable) and entry.type in scal_types:
return entry.type
elif isinstance(entry, gof.Type) and entry in scal_types:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论