Hi! I was wondering which class would you recommend to use when one defines function that takes variables or tensors as inputs and performs some transformations to enable autocompletion and etc. in IDE? There seem to be no such thing as torch.Tensor , I was trying torch.FloatTensor or _TensorBase but they seem lacking definitions of methods like torch.sum and etc. - they are defined implicitly via C extensions I guess, so pycharm can not see them.
Thanks!
upd: maybe someone built some sort of stub files 21?