!Ei\SSKJr SSKJr SSKJr SS/r"SS\5r"SS\5rg ) )Tensor)_size)ModuleFlatten Unflattenc~^\rSrSr%SrSS/r\\S'\\S'S S\S\SS4U4SjjjrS\ S\ 4S jr S\ 4S jr S r U=r$) r aS Flattens a contiguous range of dims into a tensor. For use with :class:`~nn.Sequential`, see :meth:`torch.flatten` for details. Shape: - Input: :math:`(*, S_{\text{start}},..., S_{i}, ..., S_{\text{end}}, *)`,' where :math:`S_{i}` is the size at dimension :math:`i` and :math:`*` means any number of dimensions including none. - Output: :math:`(*, \prod_{i=\text{start}}^{\text{end}} S_{i}, *)`. Args: start_dim: first dim to flatten (default = 1). end_dim: last dim to flatten (default = -1). Examples:: >>> input = torch.randn(32, 1, 5, 5) >>> # With default parameters >>> m = nn.Flatten() >>> output = m(input) >>> output.size() torch.Size([32, 25]) >>> # With non-default parameters >>> m = nn.Flatten(0, 2) >>> output = m(input) >>> output.size() torch.Size([160, 5]) start_dimend_dimreturnNc:>[TU]5 XlX lg)N)super__init__r r )selfr r __class__s U/home/mande/repo/quber/.venv/lib/python3.13/site-packages/torch/nn/modules/flatten.pyrFlatten.__init__.s " inputcNURURUR5$z Runs the forward pass. )flattenr r rrs rforwardFlatten.forward3s}}T^^T\\::rc:SURSUR3$)1 Returns the extra representation of the module. z start_dim=z , end_dim=)r r rs r extra_reprFlatten.extra_repr9s DNN+:dll^DDr)r r )r)__name__ __module__ __qualname____firstlineno____doc__ __constants__int__annotations__rrrstrr __static_attributes__ __classcell__rs@rrr sd:!),MN L#C ;V;; ECEErc^\rSrSr%Sr\\\\4rSS/r \\-\ S'\ \-\ S'S\\-S\ \-SS4U4Sjjr SSjr SS jrS \S\4S jrS\4S jrS rU=r$)r@a Unflattens a tensor dim expanding it to a desired shape. For use with :class:`~nn.Sequential`. * :attr:`dim` specifies the dimension of the input tensor to be unflattened, and it can be either `int` or `str` when `Tensor` or `NamedTensor` is used, respectively. * :attr:`unflattened_size` is the new shape of the unflattened dimension of the tensor and it can be a `tuple` of ints or a `list` of ints or `torch.Size` for `Tensor` input; a `NamedShape` (tuple of `(name, size)` tuples) for `NamedTensor` input. Shape: - Input: :math:`(*, S_{\text{dim}}, *)`, where :math:`S_{\text{dim}}` is the size at dimension :attr:`dim` and :math:`*` means any number of dimensions including none. - Output: :math:`(*, U_1, ..., U_n, *)`, where :math:`U` = :attr:`unflattened_size` and :math:`\prod_{i=1}^n U_i = S_{\text{dim}}`. Args: dim (Union[int, str]): Dimension to be unflattened unflattened_size (Union[torch.Size, Tuple, List, NamedShape]): New shape of the unflattened dimension Examples: >>> input = torch.randn(2, 50) >>> # With tuple of ints >>> m = nn.Sequential( >>> nn.Linear(50, 50), >>> nn.Unflatten(1, (2, 5, 5)) >>> ) >>> output = m(input) >>> output.size() torch.Size([2, 2, 5, 5]) >>> # With torch.Size >>> m = nn.Sequential( >>> nn.Linear(50, 50), >>> nn.Unflatten(1, torch.Size([2, 5, 5])) >>> ) >>> output = m(input) >>> output.size() torch.Size([2, 2, 5, 5]) >>> # With namedshape (tuple of tuples) >>> input = torch.randn(2, 50, names=("N", "features")) >>> unflatten = nn.Unflatten("features", (("C", 2), ("H", 5), ("W", 5))) >>> output = unflatten(input) >>> output.size() torch.Size([2, 2, 5, 5]) dimunflattened_sizer Nc>[TU]5 [U[5(aUR U5 O2[U[ 5(aUR U5 O [S5eXlX l g)Nz'invalid argument type for dim parameter) rr isinstancer)_require_tuple_intr+_require_tuple_tuple TypeErrorr1r2)rr1r2rs rrUnflatten.__init__us]  c3    # #$4 5 S ! !  % %&6 7EF F 0rc[U[5(aQ[U5HAup#[U[5(aM[SS[ U5R SU3-5e g[SS[ U5R 3-5e)Nz*unflattened_size must be tuple of tuples, but found element of type  at pos z,unflattened_size must be a tuple of tuples, zbut found type )r4tuple enumerater7typer#rridxelems rr6Unflatten._require_tuple_tuples eU # #&u- !$..#D6tDz7J7J6K8TWSXYZ.  :U 4 456 7  rc[U[[45(aQ[U5HAup#[U[5(aM[ SS[ U5RSU3-5e g[ S[ U5R35e)Nz(unflattened_size must be tuple of ints, r:r;z9unflattened_size must be a tuple of ints, but found type )r4r<listr=r)r7r>r#r?s rr5Unflatten._require_tuple_ints eeT] + +&u- !$,,#B6tDz7J7J6K8TWSXYZ. GU H\H\G] ^  rrcNURURUR5$r) unflattenr1r2rs rrUnflatten.forwardstxx)>)>??rc:SURSUR3$)rzdim=z, unflattened_size=r1r2rs rr Unflatten.extra_reprs#dhhZ243H3H2IJJrrJ)r N)r#r$r%r&r'r<r+r) NamedShaper(r*rrr6r5rrr r,r-r.s@rrr@s,\uS#X'J./M sNj(( 1C#I 19K 1PT 1    @V@@ KCKKrN) torchr torch.typesrmoduler__all__rrrrrRs; k "1Ef1EhgKgKr