Transforms#
In the spirit of torchvision and
PyG, this module contains transform operations
called on every
SpatioTemporalDataset
item get.
A transform
object expects a Data
object as input and
returns a transformed object of the same type.
Reduce graph in |
|
Rearrange all keys in |
|
Rearrange all keys in |
|
Whiten masked values in |
- class Rearrange(patterns: dict)[source]#
Rearrange all keys in
Data
according to the provided pattern using einops.rearrange.If the objects is of type
StaticBatch
, then the batch dimension in the output pattern is automatically considered.