示例#1
0
文件: __init__.py 项目: vpuhoff/CNTK
 def __init__(self, name, stream_id, storage_format, dtype, shape):
     super(StreamInformation, self).__init__()
     self.m_name = name
     self.m_id = stream_id
     self.m_storage_format = StreamInformation._storage[storage_format]
     self.m_element_type = sanitize_dtype_cntk(dtype)
     self.m_sample_layout = cntk_py.NDShape(shape)
示例#2
0
 def __init__(self, name, stream_id, storage_format, dtype, shape):
     super(StreamInformation, self).__init__()
     self.m_name = name
     self.m_id = stream_id
     self.m_storage_format = StreamInformation._storage[storage_format]
     self.m_element_type = sanitize_dtype_cntk(dtype)
     # raw NDShape is column based, so we need to reverse dimensions.
     self.m_sample_layout = cntk_py.NDShape(list(reversed(shape)))
     self.sample_shape = shape
     self.storage_format = storage_format