ignite.utils

Module with helper methods

ignite.utils.apply_to_tensor(input_, func)

Apply a function on a tensor or mapping, or sequence of tensors.

ignite.utils.apply_to_type(input_, input_type, func)

Apply a function on a object of input_type or mapping, or sequence of objects of input_type.

ignite.utils.convert_tensor(input_, device=None, non_blocking=False)

Move tensors to relevant device.

ignite.utils.to_onehot(indices, num_classes)

Convert a tensor of indices of any shape (N, …) to a tensor of one-hot indicators of shape (N, num_classes, …).