torch_npu.utils
概述
(beta)torch_npu.utils.npu_combine_tensors
(beta)torch_npu.utils.get_part_combined_tensor
(beta)torch_npu.utils.is_combined_tensor_valid
(beta)torch_npu.utils.FlopsCounter
父主题:
Ascend Extension for PyTorch自定义API