delu#

Deep Learning Utilities for PyTorch users.

to

Change devices and data types of tensors and modules in an arbitrary Python object (like torch.Tensor.to / torch.nn.Module.to, but for any Python object).

cat

Concatenate a sequence of collections of tensors (like torch.cat, but for collections of tensors).

iter_batches

Iterate over a tensor or a collection of tensors by (random) batches.