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
_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.
upd: maybe someone built some sort of stub files?