frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-21 05:34:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

Reland split unsafe version (#41484)

Browse Source 
Summary:
Reland of https://github.com/pytorch/pytorch/pull/39299

Pull Request resolved: https://github.com/pytorch/pytorch/pull/41484

Reviewed By: glaringlee

Differential Revision: D22552377

Pulled By: albanD

fbshipit-source-id: 1d1b713d2429ae162e04bda845ef0838c52df789
This commit is contained in:
Wojciech Baranowski
2020-07-16 09:00:09 -07:00
committed by Facebook GitHub Bot
parent b9442bb03e
commit 5bba973afd
17 changed files with 285 additions and 92 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
torch/_tensor_docs.py
View File

@ -3663,6 +3663,20 @@ chunk(chunks, dim=0) -> List of Tensors
See :func:`torch.chunk`
""")
add_docstr_all('unsafe_chunk',
r"""
unsafe_chunk(chunks, dim=0) -> List of Tensors
See :func:`torch.unsafe_chunk`
""")
add_docstr_all('unsafe_split',
r"""
unsafe_split(split_size, dim=0) -> List of Tensors
See :func:`torch.unsafe_split`
""")
add_docstr_all('stft',
r"""
stft(frame_length, hop, fft_size=None, return_onesided=True, window=None, pad_end=0) -> Tensor