common

TensorOrFloat

Alias for Union[float, torch.Tensor]

TensorOrInt

Alias for Union[int, torch.Tensor]

ensure_tensor(v)

convert the input to torch.Tensor if necessary.

Source code in qsparse/common.py

def ensure_tensor(v) -> torch.Tensor:
    """convert the input to `torch.Tensor` if necessary."""
    if isinstance(v, torch.Tensor):
        return v
    else:
        return torch.tensor(v)