TensorDictBase.unbind#
- TensorDictBase.unbind(dim: int) tuple[TensorDictBase, ...]#
Returns a tuple of indexed tensordicts, unbound along the indicated dimension.
The tensordicts of the result hold views of the original tensors.
- Parameters:
dim (int) – the batch dimension to remove. The result has
self.batch_size[dim]tensordicts.
Examples
>>> import torch >>> from tensordict import TensorDict >>> td = TensorDict({ ... 'x': torch.arange(12).reshape(3, 4), ... }, batch_size=[3, 4]) >>> td0, td1, td2 = td.unbind(0) >>> td0['x'] tensor([0, 1, 2, 3]) >>> td1['x'] tensor([4, 5, 6, 7])
Every tensordict class and every tensorclass has this method. For the other methods, see Methods by task.
