transform.CastTo32

class transform.CastTo32(*args: Any, **kwargs: Any)

Cast all float64 tensors to float32

Parameters:

type_map – dict with source_type: target_type (as strings)