csarofeen / pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration
http://pytorch.org
Other
26 stars 7 forks source link

Make IrCloner more convenient to use #2495

Closed zasdfgbnm closed 1 year ago

zasdfgbnm commented 1 year ago

Make IrCloner able to handle int64_t (nothing to clone, just return as is, in the future, we could add more types), std::tuple, and std::unordered_set, and their combinations recursively, for example, a set of tuple of (int64, vector). Split from https://github.com/csarofeen/pytorch/pull/2488