unsqueeze_like

pytorch_forecasting.utils.unsqueeze_like(tensor: torch.Tensor, like: torch.Tensor)

Unsqueeze last dimensions of tensor to match another tensor’s number of dimensions.

Parameters

• tensor (torch.Tensor) – tensor to unsqueeze

• like (torch.Tensor) – tensor whose dimensions to match

previous

convert_torchmetric_to_pytorch_forecasting_metric

next

FAQ