Function prepare_input_vector

model/render/render.py:254–256

(x)

    assert background is None or (background.shape[1] == resolution[0] and background.shape[2] == resolution[1])

    def prepare_input_vector(x):
        x = torch.tensor(x, dtype=torch.float32, device='cuda') if not torch.is_tensor(x) else x
        return x[:, None, None, :] if len(x.shape) == 2 else x
    
    def composite_buffer(key, layers, background, antialias):
        accum = background