funcs.stack.auto.rst.txt 993 字节
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44
stack
==========

.. currentmodule:: treetensor.torch

    
Documentation
------------------

.. autofunction:: stack

    
.. admonition:: Torch Version Related
    :class: tip

    This documentation is based on
    `torch.stack <https://pytorch.org/docs/1.12.1+cu102/generated/torch.stack.html>`_
    in `torch v1.12.1+cu102 <https://pytorch.org/docs/1.12.1+cu102/>`_.
    **Its arguments' arrangements depend on the version of pytorch you installed**.
    
    If some arguments listed here are not working properly, please check your pytorch's version 
    with the following command and find its documentation.
    
    .. code-block:: shell
        :linenos:
    
        python -c 'import torch;print(torch.__version__)'
    
    
    
    The arguments and keyword arguments supported in torch v1.12.1+cu102 is listed below.
    
    

Description From Torch v1.12.1+cu102
-----------------------------------------

.. currentmodule:: torch

    
.. autofunction:: torch.stack