[spmd expansion] support scalar_tensor (#98390)
scalar_tensor is a pure factory function that can't be handled by DTensor prop rule and needs to be currently handled in spmd expansion.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/98390
Approved by: https://github.com/mrshenli
