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.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_module

     

torch.jit.ScriptModule.register_parameter

     

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(*args, **kwargs)

     

torch.jit.ScriptModule.to(device=None, dtype=None, non_blocking=False)

     

torch.jit.ScriptModule.to(dtype, non_blocking=False)

     

torch.jit.ScriptModule.to(tensor, non_blocking=False)

     

torch.jit.ScriptModule.to(memory_format=torch.channels_last)

     

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.set_fusion_strategy

     

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