Multiple dispatch in __torch_function__

PyTorch Developer Podcast

Nội dung được cung cấp bởi PyTorch, Edward Yang, and Team PyTorch. Tất cả nội dung podcast bao gồm các tập, đồ họa và mô tả podcast đều được PyTorch, Edward Yang, and Team PyTorch hoặc đối tác nền tảng podcast của họ tải lên và cung cấp trực tiếp. Nếu bạn cho rằng ai đó đang sử dụng tác phẩm có bản quyền của bạn mà không có sự cho phép của bạn, bạn có thể làm theo quy trình được nêu ở đây https://vi.player.fm/legal.

3+ y ago 14:20

MP3•Trang chủ episode

Python is a single dispatch OO language, but there are some operations such as binary magic methods which implement a simple form of multiple dispatch. torch_function__ (through its Numpy predecessor __array_function) generalizes this mechanism so that invocations of torch.add with different subclasses work properly. This podcast describes how this mechanism works and how it can be used (in an unconventional way) to build composable subclasses ala JAX in functorch.

Further reading:

This podcast in written form https://dev-discuss.pytorch.org/t/functorch-levels-as-dynamically-allocated-classes/294
Multiple dispatch resolution rules in the RFC https://github.com/pytorch/rfcs/blob/master/RFC-0001-torch-function-for-methods.md#process-followed-during-a-functionmethod-call

83 tập

#Tech #PyTorch #Edward Yang #Team PyTorch #Deep Learning #Machine Learning