Here, we will provide a PyTorch regime to handle the partial differential equation solution of the heat equation by executing Deep Kolmogorov Method of Beck et. a' entitled as An overview on deep learning-based approximation methods for partial differential equations at https://www.aimsciences.org//article/doi/10.3934/dcdsb.2022238. The work in this code is inspired by the excellent work of Jentzen et. al. in Mathematical Introduction to Deep Learning: Methods, Implementations, and Theory.