A. Paszke, S. Gross, F. Massa, A. Lerer, J. Bradbury, G. Chanan, T. Killeen, Z. Lin,
N. Gimelshein, L. Antiga, A. Desmaison, 2019, Pytorch: An imperative style, high-performance
deep learning library, Advances in neural information processing systems, Vol. 32