axon_datasets

轴突数据集 用法 我们目前支持MNIST，FashionMNIST和CIFAR10训练数据集。 在本地下载或获取数据集： {train_images, train_labels} = AxonDatasets . MNIST . download () # Unpack train_images like... {images_binary, tensor_type, shape} = train_images 您还可以将转换函数传递给download/1 ： transform_images = fn {binary, type, shape} -> binary |> Nx . from_binary (type) |> Nx . reshape (shape) |> Nx . divide ( 255 ) |> Nx . to_batched_list (