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