提交 38c12a01 authored 作者: khaotik's avatar khaotik

vector_outs for nitsot outputs now checks array type

上级 1a42bf9b
This source diff could not be displayed because it is too large. You can view the blob instead.
...@@ -514,7 +514,8 @@ def perform( ...@@ -514,7 +514,8 @@ def perform(
if i == 0: if i == 0:
jout = j+offset_out jout = j+offset_out
shape = (store_steps[j],) + output_storage[jout].storage[0].shape shape = (store_steps[j],) + output_storage[jout].storage[0].shape
if len(output_storage[jout].storage[0].shape) == 0: if output_storage[jout].storage[0].ndim == 0 and \
isinstance(output_storage[jout].storage[0], numpy.ndarray):
vector_outs[j] = 1 vector_outs[j] = 1
dtype = output_storage[jout].storage[0].dtype dtype = output_storage[jout].storage[0].dtype
if (outs[j][0] is None or if (outs[j][0] is None or
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论