anndata.experimental.AnnLoader

class anndata.experimental.AnnLoader(adatas, batch_size=1, shuffle=False, use_default_converter=True, use_cuda=False, **kwargs)

PyTorch DataLoader用于AnnData对象。

从一系列 AnnData 对象构建 DataLoader，来自一个 AnnCollection 对象或来自一个 AnnCollectionView 对象。 处理所需的转换。

Parameters:

adatas Sequence[AnnData] | dict[str, AnnData]

AnnData 对象或一个 AnnCollection 对象，用于加载数据。

batch_size int (default: 1)

每批次加载多少样本。

shuffle bool (default: False)

设置为 True 以在每个周期重新洗牌数据。

use_default_converter bool (default: True)

使用默认转换器将数组转换为pytorch张量，传输到默认cuda设备（如果 use_cuda=True），进行内存固定（如果 pin_memory=True）。如果您传递一个带有预先指定转换器的AnnCollection对象，默认转换器将不会覆盖这些转换器，而是会在其之上应用。

use_cuda bool (default: False)

将 pytorch 张量转换后传输到默认的 cuda 设备。
仅在 use_default_converter=True 时有效

**kwargs

PyTorch DataLoader的参数。如果 adatas 不是一个 AnnCollection 对象，那么还有 AnnCollection 初始化的参数。

方法