search

nbren12 / uwnet

Neural Networks based unified physics parameterization for atmospheric models
MIT License
3 stars 2 forks source link

Add GroupBy class #34

Closed nbren12 closed 5 years ago

nbren12 commented 5 years ago

This class performs efficient groupby operations on pytorch tensors. I will need this to query SST-dependent means and scales.