unsqueeze_like#

pytorch_forecasting.utils._utils.unsqueeze_like(tensor: Tensor, like: Tensor)[来源]#

扩展张量的最后维度以匹配另一个张量的维度数。

Parameters:
  • 张量 (torch.Tensor) – 要进行扩展的张量

  • like (torch.Tensor) – 要匹配维度的张量