transform.CastTo64

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

Cast all float32 tensors to float64

Parameters:

type_map – dict with source_type: target_type (as strings)