MCPcopy Create free account

hub / github.com/pytorch/tutorials / functions

Functions760 in github.com/pytorch/tutorials

⨍ Functions 760◇ Types & classes 160↳ Endpoints 1
Functionidentity2
(x)
beginner_source/onnx/export_control_flow_model_to_onnx_tutorial.py:114
Methodis_available_while_running
(cls)
intermediate_source/ax_multiobjective_nas_tutorial.py:255
Functionmain
End-to-end CUDA graph annotation and profiling demo.
advanced_source/cuda_graph_annotations_tutorial.py:541
Functionmake_t_env
()
advanced_source/coding_ddpg.py:583
Functionmake_value_estimator
(self, value_type: ValueEstimators, **hyperparams)
advanced_source/coding_ddpg.py:247
Functionmemory_hook
(func, types, args, kwargs, result)
recipes_source/debug_mode_tutorial.py:222
Functionmy_awesome_custom_conv2d_function
(x, weight, bias, stride=[1, 1], padding=[0, 0], dilation=[1, 1], groups=1)
intermediate_source/torch_export_tutorial.py:903
Functionneg
(x)
beginner_source/onnx/export_control_flow_model_to_onnx_tutorial.py:117
Functionneg1
(x)
intermediate_source/inductor_debug_cpu.py:59
Functionneg2
(x)
intermediate_source/inductor_debug_cpu.py:178
Functionneg3
(x)
intermediate_source/inductor_debug_cpu.py:305
Functionnew_forward
(x)
beginner_source/onnx/export_control_flow_model_to_onnx_tutorial.py:113
Functionnll
(input, target)
beginner_source/nn_tutorial.py:165
Functionnumpy_sin_backward
(ctx, grad_output)
advanced_source/python_custom_ops_registrations.py:73
Functionnumpy_sin_setup_context
(ctx, inputs, output)
advanced_source/python_custom_ops_registrations.py:68
Functionnumpy_sin_strided
(x: Tensor)
advanced_source/python_custom_ops_functional.py:160
Functionnumpy_sin_strided_bad
(x: Tensor)
advanced_source/python_custom_ops_functional.py:133
Methodobservation
(self, observation)
intermediate_source/mario_rl_tutorial.py:182
Methodobservation
(self, observation)
intermediate_source/mario_rl_tutorial.py:200
Functiononnx_add_and_round
(input)
beginner_source/onnx/onnx_registry_tutorial.py:219
Functionoptimizer_hook
(parameter)
intermediate_source/optimizer_step_in_backward_tutorial.py:176
Functionpack
(x)
intermediate_source/autograd_saved_tensors_hooks_tutorial.py:228
Functionpack_hook
(x)
intermediate_source/autograd_saved_tensors_hooks_tutorial.py:167
Functionpin_copy_to_device
(*tensors)
intermediate_source/pinmem_nonblock.py:441
Functionpin_copy_to_device_nonblocking
(*tensors)
intermediate_source/pinmem_nonblock.py:448
Functionpredict_with_output_summed
(weight, bias, x)
intermediate_source/jacobians_hessians.py:290
Methodprepare_data
(self)
intermediate_source/mnist_train_nas.py:116
Functionpreprocess
(x)
beginner_source/nn_tutorial.py:751
Methodright_inverse
(self, A)
intermediate_source/parametrizations.py:279
Methodright_inverse
(self, A)
intermediate_source/parametrizations.py:306
Methodright_inverse
(self, A)
intermediate_source/parametrizations.py:344
Methodrun_node
(self, n : torch.fx.Node)
intermediate_source/fx_profiling_tutorial.py:142
Functionrun_worker
r""" A wrapper function that initializes RPC, calls the function, and shuts down RPC.
advanced_source/rpc_ddp_tutorial/main.py:112
Functionsetup
(app)
conf.py:468
Methodsetup
(self, stage=None)
intermediate_source/mnist_train_nas.py:120
Functionsetup_context
(ctx, inputs, output)
recipes_source/torch_compile_user_defined_triton_kernel_tutorial.py:258
Methodsetup_context
Store input for use in the backward pass using ``ctx.save_for_backward``. Other objects can be stored directly as attributes on the c
beginner_source/examples_autograd/polynomial_custom_function.py:41
Functionsin_kernel
( in_ptr0, out_ptr, n_elements, BLOCK_SIZE: "tl.constexpr", )
recipes_source/torch_compile_user_defined_triton_kernel_tutorial.py:216
FunctiontensorsFromPair
(pair)
intermediate_source/seq2seq_translation_tutorial.py:557
Functiontokenize_and_chunk
Tokenize text and split into fixed-length chunks for language modeling.
beginner_source/distributed_training_with_ray_tutorial.py:171
Functiontrace_handler
(p)
recipes_source/recipes/profiler_recipe.py:460
Functiontrace_handler
(p)
intermediate_source/inductor_debug_cpu.py:436
Functiontrain_cifar
(config, data_dir=None)
beginner_source/hyperparameter_tuning_tutorial.py:167
Methodtrain_dataloader
(self)
intermediate_source/mnist_train_nas.py:124
Functiontrain_func_per_worker
Training function executed by each distributed worker.
beginner_source/distributed_training_with_ray_tutorial.py:295
Functiontrainer
( log_path: str, hidden_size_1: int, hidden_size_2: int, learning_rate: float, epochs: int
intermediate_source/ax_multiobjective_nas_tutorial.py:64
Methodtraining_step
(self, batch, batch_idx)
intermediate_source/mnist_train_nas.py:97
Methodtransform_observation_spec
(self, observation_spec)
advanced_source/pendulum.py:689
Methodtransform_observation_spec
(self, observation_spec)
advanced_source/pendulum.py:713
Functiontrue_branch
(y)
intermediate_source/torch_compile_tutorial.py:375
Methodtrue_fn
(x)
intermediate_source/torch_export_tutorial.py:201
Functionunpack
(x)
intermediate_source/autograd_saved_tensors_hooks_tutorial.py:232
Functionunpack_hook
(x)
intermediate_source/autograd_saved_tensors_hooks_tutorial.py:171
Functionupdate_exp_avg_sq
(exp_avg_sq, grad, beta2)
recipes_source/foreach_map.py:87
Functionupdate_param
(param, step, exp_avg, exp_avg_sq, beta1, beta2, lr, eps)
recipes_source/foreach_map.py:90
Methodval_dataloader
(self)
intermediate_source/mnist_train_nas.py:127
Methodvalidation_step
(self, batch, batch_idx)
intermediate_source/mnist_train_nas.py:103
Functionweights_init
(m)
beginner_source/dcgan_faces_tutorial.py:292
Functionwrapper
(*args, **kwargs)
conf.py:80
Functionwrong_decomp
(x)
recipes_source/debug_mode_tutorial.py:171
← previous701–760 of 760, ranked by callers