def create_NDArrayView(shape, data_type=cntk_py.DataType_Float, dev=None):
shape = sanitize_shape(shape)
if not dev:
dev = cntk_py.DeviceDescriptor.use_default_device()
# FIXME only dense supported so far
view = cntk_py.NDArrayView(data_type, cntk_py.StorageFormat_Dense, shape,
dev)
return view
def create_NDArrayView(shape,
data_type=cntk_py.DataType_Float,
dev=cntk_device(-1)):
shape = sanitize_shape(shape)
# FIXME only dense supported so far
view = cntk_py.NDArrayView(data_type, cntk_py.StorageFormat_Dense, shape,
dev)
return view
def create_NDArrayView(shape,
data_type=cntk_py.DataType_Float,
dev=cntk_device(-1)):
if not np.isscalar(shape):
# cntk uses column major, thus we reverse the shape
shape = tuple(reversed(shape))
# FIXME only dense supported so far
view = cntk_py.NDArrayView(data_type, cntk_py.StorageFormat_Dense, shape,
dev)
return view
def create_NDArrayView_from_NumPy(nd, dev=None):
ndav_cpu = cntk_py.NDArrayView(nd, cntk_py.DeviceDescriptor.cpu_device(),
False)
if not dev:
dev = cntk_py.DeviceDescriptor.use_default_device()
ndav = ensure_dev(ndav_cpu, dev)
return ndav
def create_Value_for_Variable(var, shape=None, dev=None, mask=None):
if not dev:
dev = cntk_py.DeviceDescriptor.cpu_device()
if shape is None:
shape = var.shape().dimensions()
view = cntk_py.NDArrayView(var.get_data_type(),
cntk_py.StorageFormat_Dense, shape, dev)
if mask:
value = cntk_py.Value(view, mask)
else:
value = cntk_py.Value(view)
return value
def create_NDArrayView_from_NumPy(nd, dev=None):
if not dev:
dev = cntk_py.DeviceDescriptor.use_default_device()
view = cntk_py.NDArrayView(nd, dev, False)
return view
def create_NDArrayView_from_NumPy(nd, dev=cntk_device(-1)):
view = cntk_py.NDArrayView(nd, dev, False)
return view
def create_NDArrayView_from_NumPy(nd, dev=None):
if not dev:
dev = use_default_device()
return cntk_py.NDArrayView(nd, dev, False)
