tensorly.datasets.synthetic.gen_image

gen_image(region='swiss', image_height=20, image_width=20, n_channels=None, weight_value=1)[source]

Generates an image for regression testing

Parameters:

region{‘swiss’, ‘rectangle’}

image_heightint, optional

image_widthint, optional

weight_valuefloat, optional

n_channelsint or None, optional

if not None, the resulting image will have a third dimension

Returns:

ndarray

array of shape (image_height, image_width) or (image_height, image_width, n_channels) array for which all values are zero except the region specified