Finding number of CUDA threads, blocks, and grids for each Pytorch network model

Is it possible to get number of threads, blocks, and grid dimensions for the inference network model? if yes, how we can access them. Information like thread id, block id, grid id and SMID, and the kernel launch dimensions using Pytorch APIs (in python).