Function expand_dim

core/utils/utils.py:201–204

(tensor, dim, desired_dim_len)

    return fundamental_matrix.astype(np.float32)

def expand_dim(tensor, dim, desired_dim_len):
    sz = list(tensor.size())
    sz[dim] = desired_dim_len
    return tensor.expand(tuple(sz))


def reverse_mapping(flow, grid_map):