torch.chunk returns Tuple, not List #126278
Labels
actionable
module: typing
Related to mypy type annotations
module: viewing and reshaping
triaged
This issue has been looked at a team member, and triaged and prioritized into an appropriate module
馃摎 The doc issue
Both
claim that the return type of
torch.chunk
isList[Tensor]
, whereas in reality it isTuple[Tensor]
according toSuggest a potential alternative/fix
Fix the doc and the annotations.
cc @ezyang @malfet @xuzhao9 @gramster
The text was updated successfully, but these errors were encountered: