scglue.models.nn

Neural network modules, datasets & data loaders, and other utilities

函数

autodevice	Get torch computation device automatically based on GPU availability and memory usage
freeze_running_stats	Selectively stops normalization layers from updating running stats
get_default_numpy_dtype	Get numpy dtype matching that of the pytorch default dtype

类

GraphAttent	Graph attention
GraphConv	Graph convolution (propagation only)