Refactor center of mass calculation in DesignDatapipe to use torch te…#25

peterdsharpe merged 3 commits intoNVIDIA:mainNVIDIA/physicsnemo-cfd:mainfrom

peterdsharpe:psharpe/array-cast-fixpeterdsharpe/physicsnemo-cfd:psharpe/array-cast-fixCopy head branch name to clipboard