Block Graph Neural Networks for tumor heterogeneity prediction