提交 ebb4610f authored 作者: Matthew Rocklin's avatar Matthew Rocklin

normalize input dtypes

上级 273eaf4c
......@@ -17,7 +17,7 @@ class LoadFromDisk(Op):
@note: Non-differentiable.
"""
def __init__(self, dtype, broadcastable, mmap_mode=None):
self.dtype = dtype
self.dtype = numpy.dtype(dtype) # turn "float64" into numpy.float64
self.broadcastable = broadcastable
self.mmap_mode = mmap_mode
self._info = (dtype, broadcastable, mmap_mode)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论