spyrit.core.meas.DynamicLinearSplit.vectorize

DynamicLinearSplit.vectorize(input: tensor) → tensor

Vectorize an image-shaped tensor.

Input:

input (torch.tensor): A tensor of shape (*, h, w) where * denotes the batch size and h, w the height and width of the image.

Output:

torch.tensor: A tensor of shape (*, N) where * denotes the batch size and \(N = hw\) is the total number of pixels in the image.