提交 c9119611 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

float16 is not a valid index type.

上级 02c61708
...@@ -329,7 +329,7 @@ class Subtensor(Op): ...@@ -329,7 +329,7 @@ class Subtensor(Op):
TODO: WRITEME: This method also accepts "entry" already being a Type; TODO: WRITEME: This method also accepts "entry" already being a Type;
when would that happen? when would that happen?
""" """
invalid_scal_types = [scal.float64, scal.float32] invalid_scal_types = [scal.float64, scal.float32, scal.float16]
scal_types = [scal.int64, scal.int32, scal.int16, scal.int8] scal_types = [scal.int64, scal.int32, scal.int16, scal.int8]
tensor_types = [theano.tensor.lscalar, theano.tensor.iscalar, tensor_types = [theano.tensor.lscalar, theano.tensor.iscalar,
theano.tensor.wscalar, theano.tensor.bscalar] theano.tensor.wscalar, theano.tensor.bscalar]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论