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)
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