flatten¶
dragon.vm.torch.
flatten
(
input,
start_dim=0,
end_dim=- 1,
out=None
)[source]¶Return a tensor with dimensions flattened.
start_dim
andend_dim
could be negative:# A negative dimension is the last-k dimension x = torch.tensor([[1, 2, 3], [4, 5, 6]]) print(torch.flatten(x, start_dim=0, end_dim=1)) print(torch.flatten(x, start_dim=0, end_dim=-1)) # Equivalent
- Parameters:
- input (torch.Tensor) – The input tensor.
- start_dim (int, optional, default=0) – The start dimension to flatten.
- end_dim (int, optional, default=-1) – The end dimension to flatten.
- out (dragon.vm.torch.Tensor, optional) – The output tensor.
- Returns:
dragon.vm.torch.Tensor – The output tensor.