torch.jit

若API未标明支持情况,则代表该API的支持情况待验证。

API名称

是否支持

限制与说明

torch.jit.export

否

  

torch.jit.is_scripting

     

torch.jit.is_tracing

     

torch.jit.script

否

  

torch.jit.trace

     

torch.jit.script_if_tracing

否

  

torch.jit.trace_module

否

  

torch.jit.fork

否

  

torch.jit.wait

否

  

torch.jit.ScriptModule

否

  

torch.jit.ScriptModule.add_module

否

  

torch.jit.ScriptModule.apply

否

  

torch.jit.ScriptModule.bfloat16

否

  

torch.jit.ScriptModule.buffers

否

  

torch.jit.ScriptModule.children

否

  

torch.jit.ScriptModule.cpu

否

  

torch.jit.ScriptModule.cuda

否

  

torch.jit.ScriptModule.double

否

  

torch.jit.ScriptModule.eval

否

  

torch.jit.ScriptModule.extra_repr

否

  

torch.jit.ScriptModule.float

否

  

torch.jit.ScriptModule.get_buffer

否

  

torch.jit.ScriptModule.get_extra_state

否

  

torch.jit.ScriptModule.get_parameter

否

  

torch.jit.ScriptModule.get_submodule

否

  

torch.jit.ScriptModule.half

否

  

torch.jit.ScriptModule.ipu

否

  

torch.jit.ScriptModule.load_state_dict

否

  

torch.jit.ScriptModule.modules

否

  

torch.jit.ScriptModule.named_buffers

     

torch.jit.ScriptModule.named_children

否

  

torch.jit.ScriptModule.named_modules

否

  

torch.jit.ScriptModule.named_parameters

     

torch.jit.ScriptModule.parameters

否

  

torch.jit.ScriptModule.register_backward_hook

否

  

torch.jit.ScriptModule.register_buffer

否

  

torch.jit.ScriptModule.register_forward_hook

     

torch.jit.ScriptModule.register_forward_pre_hook

     

torch.jit.ScriptModule.register_full_backward_hook

     

torch.jit.ScriptModule.register_full_backward_pre_hook

否

  

torch.jit.ScriptModule.register_load_state_dict_post_hook

否

  

torch.jit.ScriptModule.register_module

否

  

torch.jit.ScriptModule.register_parameter

否

  

torch.jit.ScriptModule.register_state_dict_pre_hook

否

  

torch.jit.ScriptModule.requires_grad_

否

  

torch.jit.ScriptModule.save

否

  

torch.jit.ScriptModule.set_extra_state

否

  

torch.jit.ScriptModule.share_memory

否

  

torch.jit.ScriptModule.state_dict

否

  

torch.jit.ScriptModule.to

否

  

torch.jit.ScriptModule.to

否

  

torch.jit.ScriptModule.to

否

  

torch.jit.ScriptModule.to

否

  

torch.jit.ScriptModule.to

否

  

torch.jit.ScriptModule.to_empty

否

  

torch.jit.ScriptModule.train

否

  

torch.jit.ScriptModule.type

否

  

torch.jit.ScriptModule.xpu

否

  

torch.jit.ScriptModule.zero_grad

否

  

torch.jit.ScriptFunction

否

  

torch.jit.ScriptFunction.get_debug_state

否

  

torch.jit.ScriptFunction.save

否

  

torch.jit.ScriptFunction.save_to_buffer

否

  

torch.jit.freeze

否

  

torch.jit.optimize_for_inference

否

  

torch.jit.enable_onednn_fusion

否

  

torch.jit.onednn_fusion_enabled

否

  

torch.jit.set_fusion_strategy

否

  

torch.jit.strict_fusion

否

  

torch.jit.save

否

  

torch.jit.load

     

torch.jit.ignore

否

  

torch.jit.unused

否

  

torch.jit.isinstance

否

  

torch.jit.Attribute

否

  

torch.jit.Attribute.count

否

  

torch.jit.Attribute.index

否

  

torch.jit.annotate

否