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