deepinv.utils.TensorList

class deepinv.utils.TensorList(x)

Represents a list of torch.Tensor with different shapes. It allows to sum, flatten, append, etc. lists of tensors seamlessly, in a similar fashion to torch.Tensor.

Parameters:

x – a list of torch.Tensor, a single torch.Tensor or a TensorList.