ivis.models.utils.tensor_ops

Functions

format_input_tensor(input_tensor)

Format an input tensor for PyTorch's grid_sample.

ivis.models.utils.tensor_ops.format_input_tensor(input_tensor)

Format an input tensor for PyTorch’s grid_sample.

Ensures shape is (N=1, C=1, H, W), as required by grid_sample.

Parameters:

input_tensor (torch.Tensor) – A 2D, 3D, or already 4D tensor.

Returns:

formatted_tensor – Tensor reshaped for use with grid_sample.

Return type:

torch.Tensor