Skip to main content

cast_to_master

Function cast_to_master 

Source
pub fn cast_to_master(tensors: &[Tensor], master_dtype: DType) -> Vec<Tensor>
Expand description

Cast a list of tensors back to master dtype for gradient accumulation.