Dual-number dtype

I was reading the paper

PyTorch: An Imperative Style, High-Performance Deep Learning Library

and found this quote:

PyTorch can be easily extended to perform forward-mode differentiation using array-level dual numbers

Can anybody point me to an example of how to define a tensor with a dual number dtype?

Thanks

Andrea

1 Like