expand_right

class tensordict.utils.expand_right(tensor: Tensor, shape: Sequence[int])

将张量向右扩展以匹配所需的形状。

参数：

• tensor – 要扩展的张量

• shape – 目标形状

返回值：

形状与目标形状匹配的张量。

示例

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