amatorch.datasets

Load datasets and pre-trained filters.

Functions

`disparity_data`()	Load data from the 'disparity' subdirectory.
`disparity_filters`()	Load filters from the 'disparity' subdirectory.

amatorch.datasets.disparity_data()

Load data from the ‘disparity’ subdirectory.

Returns:: A dictionary containing the loaded stimuli, labels, and values. The ‘stimuli’ are reshaped to (n_samples, n_channels, n_pixels), where n_channels=2 and n_pixels=26.
Return type:: dict

amatorch.datasets.disparity_filters()

Load filters from the ‘disparity’ subdirectory.

Returns:: A tensor of shape (n_filters, n_channels, n_pixels) containing the filters, where n_channels=2 and n_pixels=26.
Return type:: torch.Tensor

Modules