rray, List[torch.FloatTensor], —
List[PIL.Image.Image], or List[np.ndarray]):
Image, NumPy array or tensor representing an image batch to be used as the starting point. For both
NumPy array and PyTorch tensor, the expected value range is between [0, 1]. If it’s a tensor or a
list or tensors, the expected shape should be (B, C, H, W) or (C, H, W). If it is a NumPy array or
a list of arrays, the expected shape should be (B, H, W, C) or (H, W, C).