expand_as_right

class tensordict.utils.expand_as_right(tensor: torch.Tensor | TensorDictBase, dest: torch.Tensor | TensorDictBase)

在右側擴充套件張量以匹配另一個張量的形狀。

引數：

• tensor – 要擴充套件的張量

• dest – 提供目標形狀的張量

返回：

形狀與 dest 輸入張量形狀匹配的張量。

示例

>>> tensor = torch.zeros(3,4)
>>> dest = torch.zeros(3,4,5)
>>> print(expand_as_right(tensor, dest).shape)
torch.Size([3,4,5])