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])