transform.CastTo32

class transform.CastTo32(*args: Any, **kwargs: Any)[source]

Cast all float64 tensors to float32

Parameters:: type_map – dict with source_type: target_type (as strings)