Give position of A relative to B, factoring in object dimensions

Saves FID statistics of one path

Builds a single .npz file from a folder of .png samples. Refer to DiT.

This function disables printing when not in master process

Shuffles a dictionary of tensors along the first dimension in unison. Example: ```python >>> x = torch.arange(6).reshape(3, 2) >

Splits a dictionary of tensors along the first dimension into `num_chunks` equal parts. Example: ```python >>> x = torch.arange(12).

Drops labels to enable classifier-free guidance.

Drops labels to enable classifier-free guidance.