Index A | B | C | D | E | F | G | H | I | J | K | L | M | N | O | P | Q | R | S | T | U | V | W | X | Z A abs() (in module tensorplay.functional) abs_() (in module tensorplay.functional) absolute() (in module tensorplay.functional) absolute_() (in module tensorplay) AcceleratorError acos() (in module tensorplay.functional) (tensorplay.SymFloat method) acos_() (in module tensorplay.functional) acosh() (in module tensorplay.functional) acosh_() (in module tensorplay.functional) adaptive_avg_pool1d() (in module tensorplay.functional) (in module tensorplay.nn.functional) adaptive_avg_pool2d() (in module tensorplay) (in module tensorplay.nn.functional) adaptive_avg_pool3d() (in module tensorplay) (in module tensorplay.nn.functional) adaptive_max_pool1d() (in module tensorplay.functional) (in module tensorplay.nn.functional) adaptive_max_pool1d_with_indices() (in module tensorplay.nn.functional) adaptive_max_pool2d() (in module tensorplay) (in module tensorplay.nn.functional) adaptive_max_pool2d_with_indices() (in module tensorplay) (in module tensorplay.nn.functional) adaptive_max_pool3d() (in module tensorplay) (in module tensorplay.nn.functional) adaptive_max_pool3d_with_indices() (in module tensorplay.nn.functional) add() (in module tensorplay.functional) add_() (in module tensorplay) add_done_callback() (tensorplay.Future method) (tensorplay.futures.Future method) add_module() (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) add_post_hook() (tensorplay.Node method) add_pre_hook() (tensorplay.Node method) add_relu() (in module tensorplay) add_safe_globals() (in module tensorplay.serialization) addbmm() (in module tensorplay.functional) addbmm_() (in module tensorplay) addcdiv() (in module tensorplay.functional) addcdiv_() (in module tensorplay) addcmul() (in module tensorplay.functional) addcmul_() (in module tensorplay) addmm() (in module tensorplay.functional) addmm_() (in module tensorplay) addmv() (in module tensorplay.functional) addmv_() (in module tensorplay.functional) addr() (in module tensorplay.functional) addr_() (in module tensorplay) adjoint() (in module tensorplay.functional) affine_grid() (in module tensorplay.nn.functional) affine_grid_generator() (in module tensorplay.functional) airy_ai() (in module tensorplay) (in module tensorplay.functional) alias() (in module tensorplay) alias_copy() (in module tensorplay.functional) all() (in module tensorplay.functional) allclose() (in module tensorplay.functional) alpha_dropout() (in module tensorplay.functional) (in module tensorplay.nn.functional) alpha_dropout_() (in module tensorplay.functional) amax() (in module tensorplay.functional) amin() (in module tensorplay.functional) aminmax() (in module tensorplay.functional) angle() (in module tensorplay.functional) any() (in module tensorplay.functional) append_cpu_child() (tensorplay.autograd.profiler_util.FunctionEvent method) apply() (tensorplay.autograd.function.Function class method) (tensorplay.autograd.function.InplaceFunction class method) (tensorplay.autograd.function.NestedIOFunction class method) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) arange() (in module tensorplay.functional) arccos() (in module tensorplay.functional) arccos_() (in module tensorplay.functional) arccosh() (in module tensorplay.functional) arccosh_() (in module tensorplay.functional) arcsin() (in module tensorplay.functional) arcsin_() (in module tensorplay.functional) arcsinh() (in module tensorplay.functional) arcsinh_() (in module tensorplay.functional) arctan() (in module tensorplay.functional) arctan2() (in module tensorplay.functional) arctan2_() (in module tensorplay) arctan_() (in module tensorplay.functional) arctanh() (in module tensorplay.functional) arctanh_() (in module tensorplay.functional) are_deterministic_algorithms_enabled() (in module tensorplay) argmax() (in module tensorplay.functional) argmin() (in module tensorplay.functional) argsort() (in module tensorplay.functional) argwhere() (in module tensorplay.functional) as_integer_ratio() (tensorplay.Layout method) (tensorplay.MemoryFormat method) (tensorplay.QScheme method) (tensorplay.SymFloat method) (tensorplay.SymInt method) as_strided() (in module tensorplay.functional) as_strided_() (in module tensorplay.functional) as_strided_copy() (in module tensorplay.functional) as_strided_scatter() (in module tensorplay.functional) as_tensor() (in module tensorplay) asarray() (in module tensorplay) asin() (in module tensorplay.functional) (tensorplay.SymFloat method) asin_() (in module tensorplay.functional) asinh() (in module tensorplay.functional) asinh_() (in module tensorplay.functional) atan() (in module tensorplay.functional) (tensorplay.SymFloat method) atan2() (in module tensorplay.functional) atan2_() (in module tensorplay) atan_() (in module tensorplay.functional) atanh() (in module tensorplay.functional) atanh_() (in module tensorplay.functional) atleast_1d() (in module tensorplay.functional) atleast_2d() (in module tensorplay.functional) atleast_3d() (in module tensorplay.functional) audio_to_tensor() (in module tensorplay) autocast (class in tensorplay) autocast_decrement_nesting() (in module tensorplay) autocast_increment_nesting() (in module tensorplay) avg_pool1d() (in module tensorplay.functional) (in module tensorplay.nn.functional) avg_pool2d() (in module tensorplay) (in module tensorplay.nn.functional) avg_pool3d() (in module tensorplay) (in module tensorplay.nn.functional) B backward() (in module tensorplay.autograd) (tensorplay.autograd.function.Function static method) (tensorplay.autograd.function.InplaceFunction static method) baddbmm() (in module tensorplay.functional) baddbmm_() (in module tensorplay) bartlett_window() (in module tensorplay.functional) batch_norm() (in module tensorplay) (in module tensorplay.nn.functional) batch_norm_backward_elemt() (in module tensorplay.functional) batch_norm_backward_reduce() (in module tensorplay.functional) batch_norm_elemt() (in module tensorplay.functional) batch_norm_gather_stats() (in module tensorplay.functional) batch_norm_gather_stats_with_counts() (in module tensorplay.functional) batch_norm_stats() (in module tensorplay.functional) batch_norm_update_stats() (in module tensorplay.functional) batch_sizes (tensorplay.nn.utils.rnn.PackedSequence attribute) BatchSampler (class in tensorplay.utils.data) begin_capture_to_if_node() (tensorplay.cuda.graphs.CUDAGraph method) begin_capture_to_while_node() (tensorplay.cuda.graphs.CUDAGraph method) bernoulli() (in module tensorplay.functional) bernoulli_() (in module tensorplay) bessel_j0() (in module tensorplay) (in module tensorplay.functional) bessel_j1() (in module tensorplay) (in module tensorplay.functional) bessel_y0() (in module tensorplay) (in module tensorplay.functional) bessel_y1() (in module tensorplay) (in module tensorplay.functional) bfloat16() (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) bilinear() (in module tensorplay.functional) (in module tensorplay.nn.functional) binary_cross_entropy() (in module tensorplay) (in module tensorplay.nn.functional) binary_cross_entropy_with_logits() (in module tensorplay) (in module tensorplay.nn.functional) bincount() (in module tensorplay.functional) binomial() (in module tensorplay.functional) bit_count() (tensorplay.Layout method) (tensorplay.MemoryFormat method) (tensorplay.QScheme method) bit_length() (tensorplay.Layout method) (tensorplay.MemoryFormat method) (tensorplay.QScheme method) (tensorplay.SymInt method) bitwise_and() (in module tensorplay.functional) bitwise_and_() (in module tensorplay) bitwise_left_shift() (in module tensorplay.functional) bitwise_left_shift_() (in module tensorplay) bitwise_not() (in module tensorplay.functional) bitwise_not_() (in module tensorplay) bitwise_or() (in module tensorplay.functional) bitwise_or_() (in module tensorplay) bitwise_right_shift() (in module tensorplay.functional) bitwise_right_shift_() (in module tensorplay) bitwise_xor() (in module tensorplay.functional) bitwise_xor_() (in module tensorplay) blackman_window() (in module tensorplay.functional) block_diag() (in module tensorplay._shape_funcs) bmm() (in module tensorplay.functional) broadcast_shapes() (in module tensorplay._shape_funcs) broadcast_tensors() (in module tensorplay._shape_funcs) broadcast_to() (in module tensorplay.functional) bucketize() (in module tensorplay.functional) buffers() (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) C caching_allocator_alloc() (in module tensorplay.cuda) caching_allocator_delete() (in module tensorplay.cuda) caching_allocator_disabled() (in module tensorplay.cuda.memory) caching_allocator_enable() (in module tensorplay.cuda.memory) calculate_gain() (in module tensorplay.nn.init) can_cast() (in module tensorplay.functional) can_device_access_peer() (in module tensorplay.cuda) capture_begin() (tensorplay.cuda.graphs.CUDAGraph method) capture_end() (tensorplay.cuda.graphs.CUDAGraph method) cartesian_prod() (in module tensorplay.functional) cat() (in module tensorplay.functional) cauchy_() (in module tensorplay) ccol_indices() (in module tensorplay) ccol_indices_copy() (in module tensorplay.functional) cdist() (in module tensorplay.functional) ceil() (in module tensorplay.functional) (tensorplay.SymFloat method) (tensorplay.SymInt method) ceil_() (in module tensorplay.functional) celu() (in module tensorplay) (in module tensorplay.nn.functional) celu_() (in module tensorplay.functional) (in module tensorplay.nn.functional) chain_matmul() (in module tensorplay.functional) ChainDataset (class in tensorplay.utils.data) change_current_allocator() (in module tensorplay.cuda) channel_shuffle() (in module tensorplay.functional) (in module tensorplay.nn.functional) chebyshev_polynomial_t() (in module tensorplay) (in module tensorplay.functional) chebyshev_polynomial_u() (in module tensorplay) (in module tensorplay.functional) chebyshev_polynomial_v() (in module tensorplay) (in module tensorplay.functional) chebyshev_polynomial_w() (in module tensorplay) (in module tensorplay.functional) check() (tensorplay.FileCheck method) check_count() (tensorplay.FileCheck method) check_dag() (tensorplay.FileCheck method) check_error() (in module tensorplay.cuda) check_next() (tensorplay.FileCheck method) check_not() (tensorplay.FileCheck method) check_regex() (tensorplay.FileCheck method) check_same() (tensorplay.FileCheck method) check_source_highlighted() (tensorplay.FileCheck method) checkpoint() (in module tensorplay.utils.checkpoint) checkpoint_sequential() (in module tensorplay.utils.checkpoint) CheckpointPolicy (class in tensorplay.utils.checkpoint) children() (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) cholesky() (in module tensorplay) cholesky_inverse() (in module tensorplay.functional) cholesky_solve() (in module tensorplay.functional) choose_qparams_optimized() (in module tensorplay.functional) chunk() (in module tensorplay.functional) circular_pad_nd() (in module tensorplay) clamp() (in module tensorplay.functional) clamp_() (in module tensorplay.functional) clamp_max() (in module tensorplay) clamp_max_() (in module tensorplay.functional) clamp_min() (in module tensorplay) clamp_min_() (in module tensorplay.functional) classproperty() (in module tensorplay.cuda) clear() (tensorplay.autograd.profiler_util.EventList method) (tensorplay.autograd.profiler_util.StringTable method) clear_autocast_cache() (in module tensorplay) clear_kernel_annotations() (in module tensorplay.cuda.graph_annotations) clear_safe_globals() (in module tensorplay.serialization) clip() (in module tensorplay.functional) clip_() (in module tensorplay.functional) clock_rate() (in module tensorplay.cuda) clone() (in module tensorplay.functional) (tensorplay.autograd.grad_mode.inference_mode method) (tensorplay.autograd.grad_mode.set_grad_enabled method) (tensorplay.SymInt method) coalesce() (in module tensorplay) col2im() (in module tensorplay) col_indices() (in module tensorplay) col_indices_copy() (in module tensorplay.functional) collect_all() (in module tensorplay.futures) column_stack() (in module tensorplay.functional) combinations() (in module tensorplay.functional) compile() (in module tensorplay._stax.api) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) complex() (in module tensorplay.functional) concat() (in module tensorplay.functional) ConcatDataset (class in tensorplay.utils.data) concatenate() (in module tensorplay.functional) conj() (in module tensorplay.functional) conj_physical() (in module tensorplay.functional) conj_physical_() (in module tensorplay.functional) conjugate() (tensorplay.Layout method) (tensorplay.MemoryFormat method) (tensorplay.QScheme method) (tensorplay.SymFloat method) (tensorplay.SymInt method) constant_() (in module tensorplay.nn.init) constant_pad_nd() (in module tensorplay) contiguous() (in module tensorplay) conv1d() (in module tensorplay.functional) (in module tensorplay.nn.functional) conv1d_grad_bias() (in module tensorplay) conv1d_grad_input() (in module tensorplay) conv1d_grad_weight() (in module tensorplay) conv2d() (in module tensorplay) (in module tensorplay.nn.functional) conv2d_grad_bias() (in module tensorplay) conv2d_grad_input() (in module tensorplay) conv2d_grad_weight() (in module tensorplay) conv2d_relu() (in module tensorplay) conv3d() (in module tensorplay.functional) (in module tensorplay.nn.functional) conv3d_grad_bias() (in module tensorplay) conv3d_grad_input() (in module tensorplay) conv3d_grad_weight() (in module tensorplay) conv_depthwise3d() (in module tensorplay) conv_tbc() (in module tensorplay.functional) (in module tensorplay.nn.functional) conv_transpose1d() (in module tensorplay.functional) (in module tensorplay.nn.functional) conv_transpose1d_grad_bias() (in module tensorplay) conv_transpose1d_grad_input() (in module tensorplay) conv_transpose1d_grad_weight() (in module tensorplay) conv_transpose2d() (in module tensorplay.functional) (in module tensorplay.nn.functional) conv_transpose2d_grad_bias() (in module tensorplay) conv_transpose2d_grad_input() (in module tensorplay) conv_transpose2d_grad_weight() (in module tensorplay) conv_transpose3d() (in module tensorplay.functional) (in module tensorplay.nn.functional) conv_transpose3d_grad_bias() (in module tensorplay) conv_transpose3d_grad_input() (in module tensorplay) conv_transpose3d_grad_weight() (in module tensorplay) convolution() (in module tensorplay.functional) convolution_backward_overrideable() (in module tensorplay) convolution_overrideable() (in module tensorplay) copy() (in module tensorplay) (tensorplay.autograd.profiler_util.EventList method) (tensorplay.autograd.profiler_util.StringTable method) copy_() (in module tensorplay) copy_sparse_to_sparse_() (in module tensorplay) copysign() (in module tensorplay.functional) copysign_() (in module tensorplay) corrcoef() (in module tensorplay.functional) cos() (in module tensorplay.functional) (tensorplay.SymFloat method) cos_() (in module tensorplay.functional) cosh() (in module tensorplay.functional) (tensorplay.SymFloat method) cosh_() (in module tensorplay.functional) cosine_embedding_loss() (in module tensorplay.functional) (in module tensorplay.nn.functional) cosine_similarity() (in module tensorplay.functional) (in module tensorplay.nn.functional) count() (tensorplay.autograd.profiler_util.EventList method) (tensorplay.autograd.profiler_util.Kernel method) (tensorplay.nn.utils.rnn.PackedSequence method) (tensorplay.Size method) count_nonzero() (in module tensorplay.functional) cov() (in module tensorplay.functional) cpu() (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) create_selective_checkpoint_contexts() (in module tensorplay.utils.checkpoint) cross() (in module tensorplay.functional) cross_entropy() (in module tensorplay.nn.functional) cross_entropy_loss() (in module tensorplay) crow_indices() (in module tensorplay) crow_indices_copy() (in module tensorplay.functional) ctc_loss() (in module tensorplay.functional) (in module tensorplay.nn.functional) cuda() (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) CudaError CUDAGraph (class in tensorplay.cuda.graphs) CUDAPluggableAllocator (class in tensorplay.cuda) cudart() (in module tensorplay.cuda) cudnn_affine_grid_generator() (in module tensorplay.functional) cudnn_batch_norm() (in module tensorplay.functional) cudnn_convolution() (in module tensorplay.functional) cudnn_convolution_add_relu() (in module tensorplay.functional) cudnn_convolution_relu() (in module tensorplay.functional) cudnn_convolution_transpose() (in module tensorplay.functional) cudnn_grid_sampler() (in module tensorplay.functional) cudnn_is_acceptable() (in module tensorplay.functional) cummax() (in module tensorplay.functional) cummin() (in module tensorplay.functional) cumprod() (in module tensorplay.functional) cumprod_() (in module tensorplay) cumsum() (in module tensorplay.functional) cumsum_() (in module tensorplay) cumulative_trapezoid() (in module tensorplay.functional) current_blas_handle() (in module tensorplay.cuda) current_device() (in module tensorplay.cuda) current_solver_handle() (in module tensorplay.cuda) current_stream() (in module tensorplay.cuda) custom_bwd() (in module tensorplay) custom_fwd() (in module tensorplay) custom_op() (in module tensorplay.library) D data (tensorplay.nn.utils.rnn.PackedSequence attribute) data() (in module tensorplay) data_ptr() (tensorplay.UntypedStorage method) DataLoader (class in tensorplay.utils.data) Dataset (class in tensorplay.utils.data) debug_dump() (tensorplay.cuda.graphs.CUDAGraph method) default_collate() (in module tensorplay.utils.data) default_convert() (in module tensorplay.utils.data) default_factory (tensorplay.autograd.profiler_util.StringTable attribute) default_generator() (in module tensorplay.random) default_restore_location() (in module tensorplay.serialization.api) default_stream() (in module tensorplay.cuda) DeferredCudaCallError define() (in module tensorplay.library) (tensorplay.library.Library method) deg2rad() (in module tensorplay.functional) deg2rad_() (in module tensorplay.functional) denominator (tensorplay.Layout attribute) (tensorplay.MemoryFormat attribute) (tensorplay.QScheme attribute) dense_dim() (in module tensorplay) dequantize() (in module tensorplay.functional) det() (in module tensorplay.functional) detach() (in module tensorplay.functional) detach_() (in module tensorplay.functional) detach_copy() (in module tensorplay.functional) detect_anomaly (class in tensorplay.autograd.anomaly_mode) Device (class in tensorplay) (class in tensorplay.cuda) device (class in tensorplay.cuda) (tensorplay.autograd.profiler_util.Kernel attribute) device_count() (in module tensorplay.cuda) device_memory_used() (in module tensorplay.cuda) device_of (class in tensorplay.cuda) DeviceMismatchError DeviceType (class in tensorplay) diag() (in module tensorplay.functional) diag_embed() (in module tensorplay.functional) diagflat() (in module tensorplay.functional) diagonal() (in module tensorplay.functional) diagonal_copy() (in module tensorplay.functional) diagonal_scatter() (in module tensorplay.functional) diff() (in module tensorplay.functional) digamma() (in module tensorplay.functional) digamma_() (in module tensorplay) dim() (in module tensorplay) dirac_() (in module tensorplay.nn.init) DispatchKey (class in tensorplay) dist() (in module tensorplay.functional) div() (in module tensorplay.functional) div_() (in module tensorplay) divide() (in module tensorplay.functional) divide_() (in module tensorplay) done() (tensorplay.Future method) (tensorplay.futures.Future method) dot() (in module tensorplay.functional) double() (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) dropout() (in module tensorplay) (in module tensorplay.nn.functional) dropout1d() (in module tensorplay.nn.functional) dropout2d() (in module tensorplay.nn.functional) dropout3d() (in module tensorplay.nn.functional) dropout_() (in module tensorplay.functional) (in module tensorplay.nn.functional) dsplit() (in module tensorplay.functional) dstack() (in module tensorplay.functional) DType (class in tensorplay) duration (tensorplay.autograd.profiler_util.Kernel attribute) E einsum() (in module tensorplay._einsum) elapsed_time() (tensorplay.cuda.streams.Event method) elapsed_us() (tensorplay.autograd.profiler_util.Interval method) elu() (in module tensorplay) (in module tensorplay.nn.functional) elu_() (in module tensorplay) (in module tensorplay.nn.functional) embedding() (in module tensorplay.functional) (in module tensorplay.nn.functional) embedding_bag() (in module tensorplay) (in module tensorplay.nn.functional) embedding_renorm_() (in module tensorplay.functional) emit_itt (class in tensorplay.profiler.profiler) emit_nvtx (class in tensorplay.profiler.profiler) empty() (in module tensorplay) empty_cache() (in module tensorplay.cuda) empty_like() (in module tensorplay.functional) empty_permuted() (in module tensorplay.functional) empty_quantized() (in module tensorplay.functional) empty_strided() (in module tensorplay.functional) enable_grad (class in tensorplay.autograd.grad_mode) end_capture_to_conditional_node() (tensorplay.cuda.graphs.CUDAGraph method) enter_dual_level() (in module tensorplay.autograd.forward_ad) entr() (in module tensorplay) (in module tensorplay.functional) eq() (in module tensorplay.functional) eq_() (in module tensorplay) equal() (in module tensorplay.functional) erf() (in module tensorplay.functional) erf_() (in module tensorplay.functional) erfc() (in module tensorplay.functional) erfc_() (in module tensorplay.functional) erfcx() (in module tensorplay) (in module tensorplay.functional) erfinv() (in module tensorplay.functional) erfinv_() (in module tensorplay) eval() (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) Event (class in tensorplay.cuda.streams) EventList (class in tensorplay.autograd.profiler_util) exit_dual_level() (in module tensorplay.autograd.forward_ad) exp() (in module tensorplay.functional) exp2() (in module tensorplay.functional) exp2_() (in module tensorplay.functional) exp_() (in module tensorplay.functional) expand() (in module tensorplay) expand_as() (in module tensorplay) expand_copy() (in module tensorplay.functional) expect_bool() (tensorplay.SymBool method) expect_float() (tensorplay.SymFloat method) expect_int() (tensorplay.SymInt method) expit() (in module tensorplay.special._error) expm1() (in module tensorplay.functional) expm1_() (in module tensorplay.functional) exponential_() (in module tensorplay) export_chrome_trace() (tensorplay.autograd.profiler_util.EventList method) export_dot() (in module tensorplay.cuda.graphs) ExternalStream (class in tensorplay.cuda.streams) extra_repr() (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) eye() (in module tensorplay.functional) eye_() (in module tensorplay.nn.init) F fake_quantize_per_channel_affine() (in module tensorplay.functional) fake_quantize_per_channel_affine_cachemask() (in module tensorplay) fake_quantize_per_tensor_affine() (in module tensorplay.functional) fake_quantize_per_tensor_affine_cachemask() (in module tensorplay) fallback() (tensorplay.library.Library method) fbgemm_linear_fp16_weight() (in module tensorplay.functional) fbgemm_linear_fp16_weight_fp32_activation() (in module tensorplay.functional) fbgemm_linear_int8_weight() (in module tensorplay.functional) fbgemm_linear_int8_weight_fp32_activation() (in module tensorplay.functional) fbgemm_linear_quantize_weight() (in module tensorplay.functional) fbgemm_pack_gemm_matrix_fp16() (in module tensorplay.functional) fbgemm_pack_quantized_matrix() (in module tensorplay.functional) feature_alpha_dropout() (in module tensorplay.functional) (in module tensorplay.nn.functional) feature_alpha_dropout_() (in module tensorplay.functional) feature_dropout() (in module tensorplay.functional) (in module tensorplay.nn.functional) feature_dropout_() (in module tensorplay.functional) (in module tensorplay.nn.functional) fft() (in module tensorplay.fft._transforms) fft2() (in module tensorplay.fft._transforms) fft_fft() (in module tensorplay) fft_fft2() (in module tensorplay) fft_fftfreq() (in module tensorplay) fft_fftn() (in module tensorplay) fft_fftshift() (in module tensorplay) fft_hfft() (in module tensorplay) fft_hfft2() (in module tensorplay) fft_hfftn() (in module tensorplay) fft_ifft() (in module tensorplay) fft_ifft2() (in module tensorplay) fft_ifftn() (in module tensorplay) fft_ifftshift() (in module tensorplay) fft_ihfft() (in module tensorplay) fft_ihfft2() (in module tensorplay) fft_ihfftn() (in module tensorplay) fft_irfft() (in module tensorplay) fft_irfft2() (in module tensorplay) fft_irfftn() (in module tensorplay) fft_rfft() (in module tensorplay) fft_rfft2() (in module tensorplay) fft_rfftfreq() (in module tensorplay) fft_rfftn() (in module tensorplay) fftfreq() (in module tensorplay.fft._frequency) fftn() (in module tensorplay.fft._transforms) fftshift() (in module tensorplay.fft._frequency) FileCheck (class in tensorplay) fill() (in module tensorplay.functional) fill_() (in module tensorplay.functional) fill_diagonal_() (in module tensorplay) finfo (class in tensorplay) fix() (in module tensorplay.functional) fix_() (in module tensorplay.functional) Flatten (class in tensorplay.nn.modules.flatten) flatten() (in module tensorplay.functional) (in module tensorplay.nn.functional) flatten_dense_tensors() (in module tensorplay) flip() (in module tensorplay.functional) fliplr() (in module tensorplay.functional) flipud() (in module tensorplay.functional) float() (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) float_power() (in module tensorplay.functional) float_power_() (in module tensorplay) floor() (in module tensorplay.functional) (tensorplay.SymFloat method) (tensorplay.SymInt method) floor_() (in module tensorplay.functional) floor_divide() (in module tensorplay.functional) floor_divide_() (in module tensorplay) fmax() (in module tensorplay.functional) fmin() (in module tensorplay.functional) fmod() (in module tensorplay.functional) fmod_() (in module tensorplay) fold() (in module tensorplay.nn.functional) fork_rng() (in module tensorplay) (in module tensorplay.random) FormattedTimesMixin (class in tensorplay.autograd.profiler_util) forward() (tensorplay.autograd.function.Function static method) (tensorplay.autograd.function.InplaceFunction static method) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) forward_add() (in module tensorplay) forward_cos() (in module tensorplay) forward_div() (in module tensorplay) forward_exp() (in module tensorplay) forward_log() (in module tensorplay) forward_mm() (in module tensorplay) forward_mul() (in module tensorplay) forward_neg() (in module tensorplay) forward_pow() (in module tensorplay) forward_relu() (in module tensorplay) forward_sigmoid() (in module tensorplay) forward_sin() (in module tensorplay) forward_sqrt() (in module tensorplay) forward_sub() (in module tensorplay) forward_tanh() (in module tensorplay) frac() (in module tensorplay.functional) frac_() (in module tensorplay.functional) fractional_max_pool2d() (in module tensorplay) (in module tensorplay.nn.functional) fractional_max_pool2d_with_indices() (in module tensorplay.nn.functional) fractional_max_pool3d() (in module tensorplay) (in module tensorplay.nn.functional) fractional_max_pool3d_with_indices() (in module tensorplay.nn.functional) frexp() (in module tensorplay.functional) frobenius_norm() (in module tensorplay) from_bytes() (tensorplay.Layout class method) (tensorplay.MemoryFormat class method) (tensorplay.QScheme class method) from_dlpack() (in module tensorplay) from_file() (in module tensorplay.functional) from_ipc_handle() (tensorplay.cuda.streams.Event class method) from_numpy() (in module tensorplay) frombuffer() (in module tensorplay) fromkeys() (tensorplay.autograd.profiler_util.StringTable class method) full() (in module tensorplay) full_like() (in module tensorplay.functional) Function (class in tensorplay.autograd.function) functional_call() (in module tensorplay.nn.utils.stateless) FunctionEvent (class in tensorplay.autograd.profiler_util) FunctionEventAvg (class in tensorplay.autograd.profiler_util) FunctionMeta (class in tensorplay.autograd.function) fused_moving_avg_obs_fake_quant() (in module tensorplay.functional) fused_mul_add() (in module tensorplay) fused_rope() (in module tensorplay) fused_swiglu() (in module tensorplay) Future (class in tensorplay) (class in tensorplay.futures) G gammainc() (in module tensorplay) (in module tensorplay.functional) gammaincc() (in module tensorplay) (in module tensorplay.functional) gammaln() (in module tensorplay.special._gamma) gather() (in module tensorplay.functional) gaussian_nll_loss() (in module tensorplay.nn.functional) gcd() (in module tensorplay.functional) gcd_() (in module tensorplay.functional) GdsFile (class in tensorplay.cuda.gds) ge() (in module tensorplay.functional) ge_() (in module tensorplay) gelu() (in module tensorplay) (in module tensorplay.nn.functional) gelu_() (in module tensorplay) Generator (class in tensorplay) geometric_() (in module tensorplay) geqrf() (in module tensorplay.functional) ger() (in module tensorplay.functional) get() (tensorplay.autograd.profiler_util.StringTable method) get_allocator_backend() (in module tensorplay.cuda) get_arch_list() (in module tensorplay.cuda) get_autocast_cpu_dtype() (in module tensorplay.amp.autocast_mode) get_autocast_dtype() (in module tensorplay) get_autocast_gpu_dtype() (in module tensorplay.amp.autocast_mode) get_backoff_factor() (tensorplay.GradScaler method) get_buffer() (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) get_crc32_options() (in module tensorplay.serialization) get_default_device() (in module tensorplay) get_default_dtype() (in module tensorplay) get_default_load_endianness() (in module tensorplay.serialization) get_default_mmap_options() (in module tensorplay.serialization) get_deterministic_debug_mode() (in module tensorplay) get_device() (in module tensorplay.functional) get_device_capability() (in module tensorplay.cuda) get_device_name() (in module tensorplay.cuda) get_device_properties() (in module tensorplay.cuda) get_extra_state() (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) get_float32_matmul_precision() (in module tensorplay) get_gencode_flags() (in module tensorplay.cuda) get_growth_factor() (tensorplay.GradScaler method) get_growth_interval() (tensorplay.GradScaler method) get_kernel() (in module tensorplay.library) get_kernel_annotations() (in module tensorplay.cuda.graph_annotations) get_num_threads() (in module tensorplay) get_numerical_jacobian_wrt_specific_input() (in module tensorplay.autograd.gradcheck) get_parallel_info() (in module tensorplay) get_parameter() (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) get_per_process_memory_fraction() (in module tensorplay.cuda) get_rng_state() (in module tensorplay) (in module tensorplay.cuda.random) (in module tensorplay.random) get_rng_state_all() (in module tensorplay.cuda.random) get_safe_globals() (in module tensorplay.serialization) get_scale() (tensorplay.GradScaler method) get_state() (tensorplay.Generator method) get_stream_from_external() (in module tensorplay.cuda) get_submodule() (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) get_sync_debug_mode() (in module tensorplay.cuda) get_thread_num() (in module tensorplay) get_unsafe_globals_in_checkpoint() (in module tensorplay.serialization) get_worker_info() (in module tensorplay.utils.data) glu() (in module tensorplay) (in module tensorplay.nn.functional) glu_backward_jvp() (in module tensorplay) glu_jvp() (in module tensorplay) grad() (in module tensorplay.autograd) gradcheck() (in module tensorplay.autograd.gradcheck) GradcheckError gradgradcheck() (in module tensorplay.autograd.gradcheck) gradient() (in module tensorplay) GradScaler (class in tensorplay) graph() (in module tensorplay.cuda.graphs) graph_pool_handle() (in module tensorplay.cuda.graphs) GraphExecGroup (class in tensorplay.utils.checkpoint) greater() (in module tensorplay.functional) greater_() (in module tensorplay) greater_equal() (in module tensorplay.functional) greater_equal_() (in module tensorplay) GreenContext (class in tensorplay.cuda.green_contexts) grid_sample() (in module tensorplay.nn.functional) grid_sampler() (in module tensorplay) grid_sampler_2d() (in module tensorplay.functional) grid_sampler_3d() (in module tensorplay.functional) group_norm() (in module tensorplay.functional) (in module tensorplay.nn.functional) grouped_mm() (in module tensorplay) (in module tensorplay.nn.functional) GroupMember (in module tensorplay.distributed.distributed_core) gru() (in module tensorplay.functional) gru_cell() (in module tensorplay.functional) (in module tensorplay.nn.functional) gt() (in module tensorplay.functional) gt_() (in module tensorplay) guard_bool() (tensorplay.SymBool method) guard_float() (tensorplay.SymFloat method) guard_int() (tensorplay.SymInt method) guard_or_false() (tensorplay.SymBool method) guard_or_true() (tensorplay.SymBool method) guard_size_oblivious() (tensorplay.SymBool method) gumbel_softmax() (in module tensorplay.nn.functional) H half() (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) hamming_window() (in module tensorplay.functional) hann_window() (in module tensorplay.functional) hardshrink() (in module tensorplay.functional) (in module tensorplay.nn.functional) hardsigmoid() (in module tensorplay) (in module tensorplay.nn.functional) hardsigmoid_() (in module tensorplay) hardswish() (in module tensorplay) (in module tensorplay.nn.functional) hardswish_() (in module tensorplay) hardtanh() (in module tensorplay) (in module tensorplay.nn.functional) hardtanh_() (in module tensorplay) (in module tensorplay.nn.functional) has_hint() (tensorplay.SymBool method) (tensorplay.SymFloat method) (tensorplay.SymInt method) has_uninitialized_params() (tensorplay.nn.modules.lazy.LazyModuleMixin method) hash_tensor() (in module tensorplay.functional) heaviside() (in module tensorplay.functional) heaviside_() (in module tensorplay) hermite_polynomial_h() (in module tensorplay) (in module tensorplay.functional) hermite_polynomial_he() (in module tensorplay) (in module tensorplay.functional) hessian() (in module tensorplay.autograd.functional) hex() (tensorplay.SymFloat method) hfft() (in module tensorplay.fft._transforms) hfft2() (in module tensorplay.fft._transforms) hfftn() (in module tensorplay.fft._transforms) hinge_embedding_loss() (in module tensorplay.functional) (in module tensorplay.nn.functional) histc() (in module tensorplay.functional) histogram() (in module tensorplay._composite_funcs) histogramdd() (in module tensorplay.functional) host_memory_stats() (in module tensorplay.cuda) host_memory_stats_as_nested_dict() (in module tensorplay.cuda) hsplit() (in module tensorplay.functional) hspmm() (in module tensorplay) hstack() (in module tensorplay.functional) huber_loss() (in module tensorplay) (in module tensorplay.nn.functional) hvp() (in module tensorplay.autograd.functional) hypot() (in module tensorplay.functional) hypot_() (in module tensorplay) I i0() (in module tensorplay.functional) i0_() (in module tensorplay.functional) i0e() (in module tensorplay) (in module tensorplay.functional) i1() (in module tensorplay) (in module tensorplay.functional) i1e() (in module tensorplay) (in module tensorplay.functional) ifft() (in module tensorplay.fft._transforms) ifft2() (in module tensorplay.fft._transforms) ifftn() (in module tensorplay.fft._transforms) ifftshift() (in module tensorplay.fft._frequency) igamma() (in module tensorplay.functional) igamma_() (in module tensorplay) igammac() (in module tensorplay.functional) igammac_() (in module tensorplay) ihfft() (in module tensorplay.fft._transforms) ihfft2() (in module tensorplay.fft._transforms) ihfftn() (in module tensorplay.fft._transforms) iinfo (class in tensorplay) im2col() (in module tensorplay) imag (tensorplay.Layout attribute) (tensorplay.MemoryFormat attribute) (tensorplay.QScheme attribute) imag() (in module tensorplay.functional) impl() (in module tensorplay.library) (tensorplay.library.Library method) impl_abstract() (in module tensorplay.library) in_interval() (tensorplay.autograd.profiler_util.MemRecordsAcc method) in_parallel_region() (in module tensorplay) index() (in module tensorplay) (tensorplay.autograd.profiler_util.EventList method) (tensorplay.autograd.profiler_util.Kernel method) (tensorplay.nn.utils.rnn.PackedSequence method) (tensorplay.Size method) index_add() (in module tensorplay.functional) index_add_() (in module tensorplay) index_copy() (in module tensorplay.functional) index_copy_() (in module tensorplay) index_fill() (in module tensorplay) index_fill_() (in module tensorplay) index_put() (in module tensorplay) index_put_() (in module tensorplay.functional) index_reduce() (in module tensorplay.functional) index_reduce_() (in module tensorplay) index_select() (in module tensorplay.functional) indices() (in module tensorplay) indices_copy() (in module tensorplay.functional) infer_schema() (in module tensorplay.library) inference_mode (class in tensorplay.autograd.grad_mode) init() (in module tensorplay.cuda) initial_seed() (in module tensorplay) (in module tensorplay.cuda.random) (in module tensorplay.random) (tensorplay.Generator method) initialize_parameters() (tensorplay.nn.modules.lazy.LazyModuleMixin method) inner() (in module tensorplay.functional) inner_backward_other() (in module tensorplay) inner_backward_self() (in module tensorplay) InplaceFunction (class in tensorplay.autograd.function) inspect_checkpoint() (in module tensorplay) instance_norm() (in module tensorplay.functional) (in module tensorplay.nn.functional) instantiate() (tensorplay.cuda.graphs.CUDAGraph method) int_repr() (in module tensorplay.functional) interpolate() (in module tensorplay.nn.functional) Interval (class in tensorplay.autograd.profiler_util) inverse() (in module tensorplay.functional) invert_permutation() (in module tensorplay.nn.utils.rnn) ipc_collect() (in module tensorplay.cuda) ipc_handle() (tensorplay.cuda.streams.Event method) irfft() (in module tensorplay.fft._transforms) irfft2() (in module tensorplay.fft._transforms) irfftn() (in module tensorplay.fft._transforms) is_autocast_available() (in module tensorplay) is_autocast_cache_enabled() (in module tensorplay) is_autocast_enabled() (in module tensorplay) is_available() (in module tensorplay.cuda) (in module tensorplay.cuda.graph_annotations) (in module tensorplay.cuda.nccl) (in module tensorplay.stax) is_bf16_supported() (in module tensorplay.cuda) is_coalesced() (in module tensorplay) is_complex() (in module tensorplay.functional) (tensorplay.Scalar method) is_conj() (in module tensorplay.functional) is_contiguous() (in module tensorplay) is_cpu() (tensorplay.cuda.Device method) (tensorplay.Device method) is_cuda (tensorplay.nn.utils.rnn.PackedSequence property) is_cuda() (tensorplay.cuda.Device method) (tensorplay.Device method) is_current_stream_capturing() (in module tensorplay.cuda.graphs) is_deterministic_algorithms_warn_only_enabled() (in module tensorplay) is_distributed() (in module tensorplay.functional) is_done() (tensorplay.Future method) (tensorplay.futures.Future method) is_enabled() (tensorplay.GradScaler method) is_floating_point() (in module tensorplay.functional) is_gds_available() (in module tensorplay.cuda) is_grad_enabled() (in module tensorplay.autograd.grad_mode) is_heap_allocated() (tensorplay.SymInt method) is_inference() (in module tensorplay.functional) is_initialized() (in module tensorplay.cuda) is_integer() (tensorplay.Layout method) (tensorplay.MemoryFormat method) (tensorplay.QScheme method) (tensorplay.SymFloat method) is_lazy() (in module tensorplay.nn.parameter) is_leaf() (in module tensorplay) is_neg() (in module tensorplay.functional) is_nonzero() (in module tensorplay.functional) is_pinned() (in module tensorplay) (tensorplay.nn.utils.rnn.PackedSequence method) is_quantized() (in module tensorplay) is_same() (tensorplay.SymInt method) is_same_size() (in module tensorplay.functional) is_set_to() (in module tensorplay) is_signed() (in module tensorplay.functional) is_storage() (in module tensorplay) is_symbolic() (tensorplay.SymBool method) (tensorplay.SymFloat method) (tensorplay.SymInt method) is_tensor() (in module tensorplay) is_tf32_supported() (in module tensorplay.cuda) is_vulkan() (tensorplay.cuda.Device method) (tensorplay.Device method) is_vulkan_available() (in module tensorplay.functional) isclose() (in module tensorplay.functional) isfinite() (in module tensorplay.functional) isin() (in module tensorplay.functional) isinf() (in module tensorplay.functional) isnan() (in module tensorplay.functional) isneginf() (in module tensorplay.functional) isposinf() (in module tensorplay.functional) isreal() (in module tensorplay.functional) istft() (in module tensorplay.functional) item() (in module tensorplay) items() (tensorplay.autograd.profiler_util.StringTable method) IterableDataset (class in tensorplay.utils.data) J jacobian() (in module tensorplay.autograd.functional) jvp() (in module tensorplay.autograd) (in module tensorplay.autograd.functional) (tensorplay.autograd.function.Function static method) (tensorplay.autograd.function.InplaceFunction static method) (tensorplay.autograd.function.NestedIOFunction static method) K kaiming_normal_() (in module tensorplay.nn.init) kaiming_uniform_() (in module tensorplay.nn.init) kaiser_window() (in module tensorplay.functional) Kernel (class in tensorplay.autograd.profiler_util) key_averages() (tensorplay.autograd.profiler_util.EventList method) keys() (tensorplay.autograd.profiler_util.StringTable method) kl_div() (in module tensorplay.functional) (in module tensorplay.nn.functional) kron() (in module tensorplay.functional) kthvalue() (in module tensorplay.functional) L l1_loss() (in module tensorplay) (in module tensorplay.nn.functional) laguerre_polynomial_l() (in module tensorplay) (in module tensorplay.functional) layer_norm() (in module tensorplay) (in module tensorplay.nn.functional) Layout (class in tensorplay) LazyModuleMixin (class in tensorplay.nn.modules.lazy) lcm() (in module tensorplay.functional) lcm_() (in module tensorplay.functional) ldexp() (in module tensorplay.functional) ldexp_() (in module tensorplay.functional) le() (in module tensorplay.functional) le_() (in module tensorplay) leaky_relu() (in module tensorplay) (in module tensorplay.nn.functional) leaky_relu_() (in module tensorplay) (in module tensorplay.nn.functional) legendre_polynomial_p() (in module tensorplay) (in module tensorplay.functional) lerp() (in module tensorplay.functional) lerp_() (in module tensorplay) less() (in module tensorplay.functional) less_() (in module tensorplay) less_equal() (in module tensorplay.functional) less_equal_() (in module tensorplay) lgamma() (in module tensorplay.functional) lgamma_() (in module tensorplay) Library (class in tensorplay.library) lift() (in module tensorplay) lift_fresh() (in module tensorplay) lift_fresh_copy() (in module tensorplay) linear() (in module tensorplay) (in module tensorplay.nn.functional) linear_cross_entropy() (in module tensorplay.nn.functional) linspace() (in module tensorplay.functional) list_gpu_processes() (in module tensorplay.cuda) load() (in module tensorplay.serialization.api) load_state_dict() (tensorplay.GradScaler method) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) local_response_norm() (in module tensorplay.nn.functional) log() (in module tensorplay.functional) log10() (in module tensorplay.functional) log10_() (in module tensorplay.functional) log1p() (in module tensorplay.functional) log1p_() (in module tensorplay.functional) log2() (in module tensorplay.functional) (tensorplay.SymFloat method) log2_() (in module tensorplay.functional) log_() (in module tensorplay.functional) log_ndtr() (in module tensorplay) (in module tensorplay.functional) log_normal_() (in module tensorplay) log_sigmoid() (in module tensorplay) log_sigmoid_forward() (in module tensorplay) log_softmax() (in module tensorplay) (in module tensorplay.nn.functional) (in module tensorplay.special._elementwise) logaddexp() (in module tensorplay.functional) logaddexp2() (in module tensorplay.functional) logcumsumexp() (in module tensorplay.functional) logdet() (in module tensorplay.functional) logical_and() (in module tensorplay.functional) logical_and_() (in module tensorplay) logical_not() (in module tensorplay.functional) logical_not_() (in module tensorplay) logical_or() (in module tensorplay.functional) logical_or_() (in module tensorplay) logical_xor() (in module tensorplay.functional) logical_xor_() (in module tensorplay) logit() (in module tensorplay.functional) logit_() (in module tensorplay.functional) logsigmoid() (in module tensorplay.nn.functional) logspace() (in module tensorplay.functional) logsumexp() (in module tensorplay.functional) (in module tensorplay.special._elementwise) lp_pool1d() (in module tensorplay.nn.functional) lp_pool2d() (in module tensorplay.nn.functional) lp_pool3d() (in module tensorplay.nn.functional) lstm() (in module tensorplay.functional) lstm_cell() (in module tensorplay.functional) (in module tensorplay.nn.functional) lt() (in module tensorplay.functional) lt_() (in module tensorplay) lu_solve() (in module tensorplay.functional) lu_unpack() (in module tensorplay.functional) M make_dual() (in module tensorplay.autograd.forward_ad) make_graphed_callables() (in module tensorplay.cuda.graphs) manual_seed() (in module tensorplay) (in module tensorplay.cuda.random) (in module tensorplay.random) (tensorplay.Generator method) manual_seed_all() (in module tensorplay.cuda.random) margin_ranking_loss() (in module tensorplay.functional) (in module tensorplay.nn.functional) mark() (in module tensorplay.cuda.nvtx) mark_kernels() (in module tensorplay.cuda.graph_annotations) masked_fill() (in module tensorplay.functional) masked_fill_() (in module tensorplay) masked_scatter() (in module tensorplay) masked_scatter_() (in module tensorplay) masked_select() (in module tensorplay.functional) matmul() (in module tensorplay.functional) matmul_backward_other() (in module tensorplay) matmul_backward_self() (in module tensorplay) matrix_exp() (in module tensorplay.functional) matrix_H() (in module tensorplay) matrix_power() (in module tensorplay.functional) max() (in module tensorplay) (tensorplay.SymFloat method) (tensorplay.SymInt method) max_memory_allocated() (in module tensorplay.cuda) max_memory_reserved() (in module tensorplay.cuda) max_pool1d() (in module tensorplay.functional) (in module tensorplay.nn.functional) max_pool1d_with_indices() (in module tensorplay) (in module tensorplay.nn.functional) max_pool2d() (in module tensorplay) (in module tensorplay.nn.functional) max_pool2d_with_indices() (in module tensorplay) (in module tensorplay.nn.functional) max_pool3d() (in module tensorplay.functional) (in module tensorplay.nn.functional) max_pool3d_with_indices() (in module tensorplay) (in module tensorplay.nn.functional) max_unpool1d() (in module tensorplay.nn.functional) max_unpool2d() (in module tensorplay) (in module tensorplay.nn.functional) max_unpool3d() (in module tensorplay) (in module tensorplay.nn.functional) maximum() (in module tensorplay.functional) maybe_as_bool() (tensorplay.SymBool method) maybe_as_float() (tensorplay.SymFloat method) maybe_as_int() (tensorplay.SymInt method) mean() (in module tensorplay.functional) median() (in module tensorplay.functional) mem_get_info() (in module tensorplay.cuda) memory_allocated() (in module tensorplay.cuda) memory_reserved() (in module tensorplay.cuda) memory_snapshot() (in module tensorplay.cuda) memory_stats() (in module tensorplay.cuda) memory_stats_as_nested_dict() (in module tensorplay.cuda) memory_summary() (in module tensorplay.cuda) memory_usage() (in module tensorplay.cuda) MemoryFormat (class in tensorplay) MemPool (class in tensorplay.cuda) MemRecordsAcc (class in tensorplay.autograd.profiler_util) meshgrid() (in module tensorplay.functional) mH() (in module tensorplay) min() (in module tensorplay) (tensorplay.SymFloat method) (tensorplay.SymInt method) minimum() (in module tensorplay.functional) miopen_batch_norm() (in module tensorplay.functional) miopen_convolution() (in module tensorplay.functional) miopen_convolution_add_relu() (in module tensorplay.functional) miopen_convolution_relu() (in module tensorplay.functional) miopen_convolution_transpose() (in module tensorplay.functional) miopen_ctc_loss() (in module tensorplay.functional) miopen_depthwise_convolution() (in module tensorplay.functional) miopen_rnn() (in module tensorplay.functional) mish() (in module tensorplay) (in module tensorplay.nn.functional) mish_() (in module tensorplay) mkldnn_adaptive_avg_pool2d() (in module tensorplay.functional) mkldnn_convolution() (in module tensorplay.functional) mkldnn_linear() (in module tensorplay) mkldnn_linear_backward_input() (in module tensorplay) mkldnn_linear_backward_weights() (in module tensorplay.functional) mkldnn_max_pool2d() (in module tensorplay.functional) mkldnn_max_pool3d() (in module tensorplay.functional) mkldnn_reorder_conv2d_weight() (in module tensorplay) mkldnn_reorder_conv3d_weight() (in module tensorplay) mkldnn_rnn_layer() (in module tensorplay.functional) mm() (in module tensorplay.functional) mode() (in module tensorplay.functional) modified_bessel_i0() (in module tensorplay) (in module tensorplay.functional) modified_bessel_i1() (in module tensorplay) (in module tensorplay.functional) modified_bessel_k0() (in module tensorplay) (in module tensorplay.functional) modified_bessel_k1() (in module tensorplay) (in module tensorplay.functional) module tensorplay.amp.autocast_mode tensorplay.amp.grad_scaler tensorplay.cpu.amp tensorplay.cpu.amp.autocast_mode tensorplay.cpu.amp.grad_scaler tensorplay.cuda.amp tensorplay.cuda.amp.autocast_mode tensorplay.cuda.amp.common tensorplay.cuda.amp.grad_scaler tensorplay.distributed.algorithms tensorplay.distributed.algorithms.ddp_comm_hooks tensorplay.distributed.algorithms.ddp_comm_hooks.ddp_zero_hook tensorplay.distributed.algorithms.ddp_comm_hooks.debugging_hooks tensorplay.distributed.algorithms.ddp_comm_hooks.default_hooks tensorplay.distributed.algorithms.ddp_comm_hooks.mixed_precision_hooks tensorplay.distributed.algorithms.ddp_comm_hooks.optimizer_overlap_hooks tensorplay.distributed.algorithms.ddp_comm_hooks.post_localSGD_hook tensorplay.distributed.algorithms.ddp_comm_hooks.powerSGD_hook tensorplay.distributed.algorithms.ddp_comm_hooks.quantization_hooks tensorplay.distributed.algorithms.join tensorplay.distributed.algorithms.model_averaging tensorplay.distributed.algorithms.model_averaging.averagers tensorplay.distributed.algorithms.model_averaging.hierarchical_model_averager tensorplay.distributed.algorithms.model_averaging.utils tensorplay.distributed.argparse_util tensorplay.distributed.checkpoint.api tensorplay.distributed.checkpoint.default_planner tensorplay.distributed.checkpoint.filesystem tensorplay.distributed.checkpoint.hf_storage tensorplay.distributed.checkpoint.metadata tensorplay.distributed.checkpoint.optimizer tensorplay.distributed.checkpoint.planner tensorplay.distributed.checkpoint.planner_helpers tensorplay.distributed.checkpoint.quantized_hf_storage tensorplay.distributed.checkpoint.resharding tensorplay.distributed.checkpoint.state_dict tensorplay.distributed.checkpoint.state_dict_loader tensorplay.distributed.checkpoint.state_dict_saver tensorplay.distributed.checkpoint.stateful tensorplay.distributed.checkpoint.storage tensorplay.distributed.checkpoint.utils tensorplay.distributed.constants tensorplay.distributed.device_mesh tensorplay.distributed.distributed_core tensorplay.distributed.elastic tensorplay.distributed.elastic.agent.server.api tensorplay.distributed.elastic.agent.server.local_elastic_agent tensorplay.distributed.elastic.events.api tensorplay.distributed.elastic.events.handlers tensorplay.distributed.elastic.metrics.api tensorplay.distributed.elastic.multiprocessing.api tensorplay.distributed.elastic.multiprocessing.errors.error_handler tensorplay.distributed.elastic.multiprocessing.errors.handlers tensorplay.distributed.elastic.multiprocessing.redirects tensorplay.distributed.elastic.multiprocessing.tail_log tensorplay.distributed.elastic.rendezvous.api tensorplay.distributed.elastic.rendezvous.c10d_rendezvous_backend tensorplay.distributed.elastic.rendezvous.dynamic_rendezvous tensorplay.distributed.elastic.rendezvous.etcd_rendezvous tensorplay.distributed.elastic.rendezvous.etcd_rendezvous_backend tensorplay.distributed.elastic.rendezvous.etcd_server tensorplay.distributed.elastic.rendezvous.etcd_store tensorplay.distributed.elastic.rendezvous.static_tcp_rendezvous tensorplay.distributed.elastic.rendezvous.utils tensorplay.distributed.elastic.timer.api tensorplay.distributed.elastic.timer.file_based_local_timer tensorplay.distributed.elastic.timer.local_timer tensorplay.distributed.elastic.utils tensorplay.distributed.elastic.utils.api tensorplay.distributed.elastic.utils.data tensorplay.distributed.elastic.utils.data.cycling_iterator tensorplay.distributed.elastic.utils.data.elastic_distributed_sampler tensorplay.distributed.elastic.utils.distributed tensorplay.distributed.elastic.utils.log_level tensorplay.distributed.elastic.utils.logging tensorplay.distributed.elastic.utils.store tensorplay.distributed.fsdp.api tensorplay.distributed.fsdp.fully_sharded_data_parallel tensorplay.distributed.fsdp.sharded_grad_scaler tensorplay.distributed.fsdp.wrap tensorplay.distributed.launcher tensorplay.distributed.logging_handlers tensorplay.distributed.nn tensorplay.distributed.nn.api tensorplay.distributed.nn.api.remote_module tensorplay.distributed.nn.functional tensorplay.distributed.nn.jit tensorplay.distributed.nn.jit.instantiator tensorplay.distributed.nn.jit.templates tensorplay.distributed.nn.jit.templates.remote_module_template tensorplay.distributed.optim.apply_optimizer_in_backward tensorplay.distributed.optim.functional_adadelta tensorplay.distributed.optim.functional_adagrad tensorplay.distributed.optim.functional_adam tensorplay.distributed.optim.functional_adamax tensorplay.distributed.optim.functional_adamw tensorplay.distributed.optim.functional_rmsprop tensorplay.distributed.optim.functional_rprop tensorplay.distributed.optim.functional_sgd tensorplay.distributed.optim.named_optimizer tensorplay.distributed.optim.optimizer tensorplay.distributed.optim.post_localSGD_optimizer tensorplay.distributed.optim.utils tensorplay.distributed.optim.zero_redundancy_optimizer tensorplay.distributed.p10d_logger tensorplay.distributed.remote_device tensorplay.distributed.rendezvous tensorplay.distributed.rpc.api tensorplay.distributed.rpc.backend_registry tensorplay.distributed.rpc.constants tensorplay.distributed.rpc.functions tensorplay.distributed.rpc.internal tensorplay.distributed.rpc.options tensorplay.distributed.rpc.rref_proxy tensorplay.distributed.rpc.server_process_global_profiler tensorplay.distributed.tensor.parallel.api tensorplay.distributed.tensor.parallel.ddp tensorplay.distributed.tensor.parallel.fsdp tensorplay.distributed.tensor.parallel.input_reshard tensorplay.distributed.tensor.parallel.loss tensorplay.distributed.tensor.parallel.style tensorplay.distributed.utils tensorplay.multiprocessing.pool tensorplay.multiprocessing.queue tensorplay.multiprocessing.reductions tensorplay.nn.backends tensorplay.nn.backends.thnn tensorplay.nn.common_types tensorplay.nn.cpp tensorplay.nn.functional tensorplay.nn.grad tensorplay.nn.init tensorplay.nn.modules.activation tensorplay.nn.modules.adaptive tensorplay.nn.modules.batchnorm tensorplay.nn.modules.channelshuffle tensorplay.nn.modules.container tensorplay.nn.modules.conv tensorplay.nn.modules.distance tensorplay.nn.modules.dropout tensorplay.nn.modules.flatten tensorplay.nn.modules.folding tensorplay.nn.modules.instancenorm tensorplay.nn.modules.lazy tensorplay.nn.modules.linear tensorplay.nn.modules.loss tensorplay.nn.modules.module tensorplay.nn.modules.normalization tensorplay.nn.modules.padding tensorplay.nn.modules.pixelshuffle tensorplay.nn.modules.pooling tensorplay.nn.modules.rnn tensorplay.nn.modules.sparse tensorplay.nn.modules.transformer tensorplay.nn.modules.upsampling tensorplay.nn.modules.utils tensorplay.nn.parallel.comm tensorplay.nn.parallel.distributed tensorplay.nn.parallel.parallel_apply tensorplay.nn.parallel.replicate tensorplay.nn.parallel.scatter_gather tensorplay.nn.parameter tensorplay.nn.utils.clip_grad tensorplay.nn.utils.convert_parameters tensorplay.nn.utils.fusion tensorplay.nn.utils.init tensorplay.nn.utils.memory_format tensorplay.nn.utils.parametrizations tensorplay.nn.utils.parametrize tensorplay.nn.utils.prune tensorplay.nn.utils.rnn tensorplay.nn.utils.stateless tensorplay.optim.lr_scheduler tensorplay.optim.optimizer tensorplay.optim.swa_utils tensorplay.utils.data.datapipes tensorplay.utils.data.datapipes.dataframe tensorplay.utils.data.datapipes.iter tensorplay.utils.data.datapipes.map tensorplay.utils.data.datapipes.utils tensorplay.utils.serialization tensorplay.utils.serialization.config modules() (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) moveaxis() (in module tensorplay.functional) movedim() (in module tensorplay.functional) mro() (tensorplay.autograd.function.FunctionMeta method) (tensorplay.autograd.variable.VariableMeta method) mse_loss() (in module tensorplay) (in module tensorplay.nn.functional) msort() (in module tensorplay.functional) mT() (in module tensorplay) mul() (in module tensorplay.functional) mul_() (in module tensorplay) multi_head_attention_forward() (in module tensorplay.nn.functional) multi_margin_loss() (in module tensorplay) (in module tensorplay.nn.functional) multigammaln() (in module tensorplay.special._gamma) multilabel_margin_loss() (in module tensorplay) (in module tensorplay.nn.functional) multilabel_margin_loss_forward() (in module tensorplay) multilabel_soft_margin_loss() (in module tensorplay) (in module tensorplay.nn.functional) multinomial() (in module tensorplay.functional) multiply() (in module tensorplay.functional) multiply_() (in module tensorplay) mv() (in module tensorplay.functional) mvlgamma() (in module tensorplay.functional) mvlgamma_() (in module tensorplay) N name (tensorplay.autograd.profiler_util.Kernel attribute) named_buffers() (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) named_children() (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) named_modules() (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) named_parameters() (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) NamedShape (tensorplay.nn.modules.flatten.Unflatten attribute) nan_to_num() (in module tensorplay.functional) nan_to_num_() (in module tensorplay.functional) nanmean() (in module tensorplay.functional) nanmedian() (in module tensorplay.functional) nanquantile() (in module tensorplay._composite_funcs) nansum() (in module tensorplay.functional) narrow() (in module tensorplay.functional) narrow_copy() (in module tensorplay.functional) native_alpha_dropout() (in module tensorplay) native_batch_norm() (in module tensorplay.functional) native_channel_shuffle() (in module tensorplay.functional) (in module tensorplay.nn.functional) native_dropout() (in module tensorplay) native_feature_dropout() (in module tensorplay) native_group_norm() (in module tensorplay.functional) native_layer_norm() (in module tensorplay.functional) native_norm() (in module tensorplay.functional) nbytes() (tensorplay.UntypedStorage method) ndtr() (in module tensorplay) (in module tensorplay.functional) ndtri() (in module tensorplay) (in module tensorplay.functional) ne() (in module tensorplay.functional) ne_() (in module tensorplay) neg() (in module tensorplay.functional) neg_() (in module tensorplay.functional) negative() (in module tensorplay.functional) negative_() (in module tensorplay.functional) nested_to_padded_tensor() (in module tensorplay) NestedIOFunction (class in tensorplay.autograd.function) new_empty() (in module tensorplay) new_empty_strided() (in module tensorplay) new_full() (in module tensorplay) new_ones() (in module tensorplay) new_zeros() (in module tensorplay) nextafter() (in module tensorplay.functional) nextafter_() (in module tensorplay) nll_loss() (in module tensorplay) (in module tensorplay.nn.functional) nll_loss2d() (in module tensorplay) nll_loss2d_forward() (in module tensorplay) nll_loss_forward() (in module tensorplay) nll_loss_nd() (in module tensorplay) no_grad (class in tensorplay.autograd.grad_mode) Node (class in tensorplay) nonzero() (in module tensorplay.functional) nonzero_numpy() (in module tensorplay) nonzero_static() (in module tensorplay.functional) norm() (in module tensorplay.functional) norm_except_dim() (in module tensorplay.functional) normal() (in module tensorplay.functional) normal_() (in module tensorplay) (in module tensorplay.nn.init) normal_functional() (in module tensorplay) normalize() (in module tensorplay.nn.functional) not_equal() (in module tensorplay.functional) not_equal_() (in module tensorplay) nuclear_norm() (in module tensorplay.functional) numel() (in module tensorplay.functional) (tensorplay.Size method) numerator (tensorplay.Layout attribute) (tensorplay.MemoryFormat attribute) (tensorplay.QScheme attribute) numpy_T() (in module tensorplay) O once_differentiable() (in module tensorplay.autograd.function) one_hot() (in module tensorplay) (in module tensorplay.nn.functional) ones() (in module tensorplay) ones_() (in module tensorplay.nn.init) ones_like() (in module tensorplay.functional) opcheck() (in module tensorplay.library) orgqr() (in module tensorplay.functional) ormqr() (in module tensorplay.functional) orthogonal_() (in module tensorplay.nn.init) outer() (in module tensorplay.functional) OutOfMemoryError output_nr() (in module tensorplay) P pack_padded_sequence() (in module tensorplay.nn.utils.rnn) pack_sequence() (in module tensorplay.nn.utils.rnn) PackedSequence (class in tensorplay.nn.utils.rnn) pad() (in module tensorplay) (in module tensorplay.nn.functional) pad_packed_sequence() (in module tensorplay.nn.utils.rnn) pad_sequence() (in module tensorplay) (in module tensorplay.nn.utils.rnn) pairwise_distance() (in module tensorplay.functional) (in module tensorplay.nn.functional) parameters() (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) pdist() (in module tensorplay.functional) (in module tensorplay.nn.functional) permute() (in module tensorplay.functional) permute_copy() (in module tensorplay.functional) pin_memory() (in module tensorplay) pinverse() (in module tensorplay.functional) pixel_shuffle() (in module tensorplay) (in module tensorplay.nn.functional) pixel_unshuffle() (in module tensorplay.functional) (in module tensorplay.nn.functional) poisson() (in module tensorplay.functional) poisson_nll_loss() (in module tensorplay.functional) (in module tensorplay.nn.functional) polar() (in module tensorplay.functional) polygamma() (in module tensorplay.functional) (in module tensorplay.special._gamma) polygamma_() (in module tensorplay) pool_id (tensorplay.cuda.graphs.CUDAGraph property) pop() (tensorplay.autograd.profiler_util.EventList method) (tensorplay.autograd.profiler_util.StringTable method) popitem() (tensorplay.autograd.profiler_util.StringTable method) positive() (in module tensorplay.functional) pow() (in module tensorplay.functional) pow_() (in module tensorplay) pow_by_natural() (tensorplay.SymInt method) power_draw() (in module tensorplay.cuda) prelu() (in module tensorplay.functional) (in module tensorplay.nn.functional) prod() (in module tensorplay.functional) profile (class in tensorplay.profiler.profiler) profile() (in module tensorplay.cuda.profiler) promote_types() (in module tensorplay.functional) psi() (in module tensorplay.special._gamma) put() (in module tensorplay.functional) put_() (in module tensorplay) Q q_per_channel_axis() (in module tensorplay.functional) q_per_channel_scales() (in module tensorplay.functional) q_per_channel_zero_points() (in module tensorplay.functional) q_scale() (in module tensorplay.functional) q_zero_point() (in module tensorplay.functional) QScheme (class in tensorplay) qscheme() (in module tensorplay) quantile() (in module tensorplay._composite_funcs) quantize_per_channel() (in module tensorplay.functional) quantize_per_tensor() (in module tensorplay.functional) quantize_per_tensor_dynamic() (in module tensorplay.functional) quantized_add() (in module tensorplay) quantized_batch_norm() (in module tensorplay.functional) quantized_clamp() (in module tensorplay) quantized_conv2d() (in module tensorplay) quantized_conv2d_prepack() (in module tensorplay) quantized_conv2d_run() (in module tensorplay) quantized_conv2d_unpack() (in module tensorplay) quantized_div() (in module tensorplay) quantized_gru_cell() (in module tensorplay.functional) quantized_linear() (in module tensorplay) quantized_lstm_cell() (in module tensorplay.functional) quantized_max_pool1d() (in module tensorplay.functional) quantized_max_pool2d() (in module tensorplay.functional) quantized_max_pool3d() (in module tensorplay.functional) quantized_mul() (in module tensorplay) quantized_rnn_relu_cell() (in module tensorplay.functional) quantized_rnn_tanh_cell() (in module tensorplay.functional) quantized_sub() (in module tensorplay) query() (tensorplay.cuda.streams.Event method) (tensorplay.cuda.streams.ExternalStream method) (tensorplay.cuda.streams.Stream method) R rad2deg() (in module tensorplay.functional) rad2deg_() (in module tensorplay.functional) rand() (in module tensorplay) rand_like() (in module tensorplay.functional) randint() (in module tensorplay.functional) randint_like() (in module tensorplay.functional) randn() (in module tensorplay) randn_like() (in module tensorplay.functional) random_() (in module tensorplay) RandomSampler (class in tensorplay.utils.data) randperm() (in module tensorplay.functional) range() (in module tensorplay.cuda.nvtx) (in module tensorplay.functional) range_end() (in module tensorplay.cuda.nvtx) range_pop() (in module tensorplay.cuda.nvtx) range_push() (in module tensorplay.cuda.nvtx) range_start() (in module tensorplay.cuda.nvtx) ravel() (in module tensorplay.functional) real (tensorplay.Layout attribute) (tensorplay.MemoryFormat attribute) (tensorplay.QScheme attribute) real() (in module tensorplay.functional) reciprocal() (in module tensorplay.functional) reciprocal_() (in module tensorplay.functional) record() (tensorplay.cuda.streams.Event method) record_event() (tensorplay.cuda.streams.ExternalStream method) (tensorplay.cuda.streams.Stream method) record_function (class in tensorplay.profiler.profiler) reduce_op (built-in class) reflection_pad1d() (in module tensorplay) reflection_pad2d() (in module tensorplay) reflection_pad3d() (in module tensorplay) reflection_pad_nd() (in module tensorplay) register_autocast() (in module tensorplay.library) register_autograd() (in module tensorplay.library) register_buffer() (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) register_fake() (in module tensorplay.library) register_forward_hook() (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) register_forward_pre_hook() (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) register_full_backward_hook() (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) register_full_backward_pre_hook() (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) register_kernel() (in module tensorplay.library) register_load_state_dict_post_hook() (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) register_load_state_dict_pre_hook() (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) register_module() (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) register_module_buffer_registration_hook() (in module tensorplay.nn.modules.module) register_module_forward_hook() (in module tensorplay.nn.modules.module) register_module_forward_pre_hook() (in module tensorplay.nn.modules.module) register_module_full_backward_hook() (in module tensorplay.nn.modules.module) register_module_full_backward_pre_hook() (in module tensorplay.nn.modules.module) register_module_module_registration_hook() (in module tensorplay.nn.modules.module) register_module_parameter_registration_hook() (in module tensorplay.nn.modules.module) register_package() (in module tensorplay.serialization) register_parameter() (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) register_state_dict_post_hook() (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) register_state_dict_pre_hook() (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) register_vmap() (in module tensorplay.library) relu() (in module tensorplay) (in module tensorplay.nn.functional) relu6() (in module tensorplay) (in module tensorplay.nn.functional) relu6_() (in module tensorplay) relu_() (in module tensorplay.functional) (in module tensorplay.nn.functional) remainder() (in module tensorplay.functional) remainder_() (in module tensorplay) remove() (tensorplay.autograd.profiler_util.EventList method) renorm() (in module tensorplay.functional) renorm_() (in module tensorplay) repeat() (in module tensorplay) repeat_interleave() (in module tensorplay.functional) replay() (tensorplay.cuda.graphs.CUDAGraph method) replication_pad1d() (in module tensorplay) replication_pad2d() (in module tensorplay) replication_pad3d() (in module tensorplay) replication_pad_nd() (in module tensorplay) requires_grad_() (in module tensorplay) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) reset() (tensorplay.cuda.graphs.CUDAGraph method) reset_accumulated_host_memory_stats() (in module tensorplay.cuda) reset_accumulated_memory_stats() (in module tensorplay.cuda) reset_peak_host_memory_stats() (in module tensorplay.cuda) reset_peak_memory_stats() (in module tensorplay.cuda) reshape() (in module tensorplay.functional) reshape_as() (in module tensorplay) resizable() (tensorplay.UntypedStorage method) resize_() (in module tensorplay) (tensorplay.UntypedStorage method) resize_as_() (in module tensorplay.functional) resize_as_sparse_() (in module tensorplay.functional) resolve_conj() (in module tensorplay.functional) resolve_neg() (in module tensorplay.functional) result_type() (in module tensorplay.functional) retain_grad() (in module tensorplay) retains_grad() (in module tensorplay) reverse() (tensorplay.autograd.profiler_util.EventList method) rfft() (in module tensorplay.fft._transforms) rfft2() (in module tensorplay.fft._transforms) rfftfreq() (in module tensorplay.fft._frequency) rfftn() (in module tensorplay.fft._transforms) rms_norm() (in module tensorplay.functional) (in module tensorplay.nn.functional) rnn_relu() (in module tensorplay.functional) rnn_relu_cell() (in module tensorplay.functional) (in module tensorplay.nn.functional) rnn_tanh() (in module tensorplay.functional) rnn_tanh_cell() (in module tensorplay.functional) (in module tensorplay.nn.functional) roll() (in module tensorplay.functional) rot90() (in module tensorplay.functional) rotary_embedding() (in module tensorplay) round() (in module tensorplay.functional) (in module tensorplay.special._elementwise) (tensorplay.SymFloat method) (tensorplay.SymInt method) round_() (in module tensorplay.functional) row_indices() (in module tensorplay) row_indices_copy() (in module tensorplay.functional) row_stack() (in module tensorplay.functional) rrelu() (in module tensorplay.functional) (in module tensorplay.nn.functional) rrelu_() (in module tensorplay.functional) (in module tensorplay.nn.functional) rrelu_with_noise() (in module tensorplay) rrelu_with_noise_() (in module tensorplay) rsqrt() (in module tensorplay.functional) rsqrt_() (in module tensorplay.functional) rsub() (in module tensorplay.functional) run() (tensorplay.FileCheck method) S safe_globals (class in tensorplay.serialization) sample() (in module tensorplay) Sampler (class in tensorplay.utils.data) save() (in module tensorplay.serialization.api) save_on_cpu() (in module tensorplay.autograd.graph) saved_tensors_hooks() (in module tensorplay.autograd.graph) Scalar (class in tensorplay) scalar_tensor() (in module tensorplay.functional) scale() (tensorplay.GradScaler method) scaled_dot_product_attention() (in module tensorplay) (in module tensorplay.nn.functional) scaled_grouped_mm() (in module tensorplay.nn.functional) scaled_mm() (in module tensorplay.nn.functional) scaled_modified_bessel_k0() (in module tensorplay) (in module tensorplay.functional) scaled_modified_bessel_k1() (in module tensorplay) (in module tensorplay.functional) scatter() (in module tensorplay.functional) scatter_() (in module tensorplay) scatter_add() (in module tensorplay.functional) scatter_add_() (in module tensorplay) scatter_reduce() (in module tensorplay.functional) scatter_reduce_() (in module tensorplay) searchsorted() (in module tensorplay.functional) seed() (in module tensorplay) (in module tensorplay.cuda.random) (in module tensorplay.random) (tensorplay.Generator method) seed_all() (in module tensorplay.cuda.random) segment_reduce() (in module tensorplay.functional) select() (in module tensorplay.functional) select_copy() (in module tensorplay.functional) select_scatter() (in module tensorplay.functional) SelectiveCheckpointContext (class in tensorplay.utils.checkpoint) selu() (in module tensorplay.functional) (in module tensorplay.nn.functional) selu_() (in module tensorplay.functional) (in module tensorplay.nn.functional) SequentialSampler (class in tensorplay.utils.data) set_() (in module tensorplay) set_autocast_cache_enabled() (in module tensorplay) set_autocast_dtype() (in module tensorplay) set_autocast_enabled() (in module tensorplay) set_backoff_factor() (tensorplay.GradScaler method) set_checkpoint_debug_enabled() (in module tensorplay.utils.checkpoint) set_checkpoint_early_stop() (in module tensorplay.utils.checkpoint) set_conditional_handle_for_current_node() (tensorplay.cuda.graphs.CUDAGraph method) set_cpu_parent() (tensorplay.autograd.profiler_util.FunctionEvent method) set_crc32_options() (in module tensorplay.serialization) set_data() (in module tensorplay) set_default_device() (in module tensorplay) set_default_dtype() (in module tensorplay) set_default_load_endianness() (in module tensorplay.serialization) set_default_mmap_options() (in module tensorplay.serialization) set_detect_anomaly (class in tensorplay.autograd.anomaly_mode) set_deterministic_debug_mode() (in module tensorplay) set_device() (in module tensorplay.cuda) set_device_states() (in module tensorplay.utils.checkpoint) set_exception() (tensorplay.Future method) set_extra_state() (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) set_float32_matmul_precision() (in module tensorplay) set_grad_enabled (class in tensorplay.autograd.grad_mode) set_growth_factor() (tensorplay.GradScaler method) set_growth_interval() (tensorplay.GradScaler method) set_num_threads() (in module tensorplay) set_per_process_memory_fraction() (in module tensorplay.cuda) set_printoptions() (in module tensorplay) set_result() (tensorplay.Future method) (tensorplay.futures.Future method) set_rng_state() (in module tensorplay) (in module tensorplay.cuda.random) (in module tensorplay.random) set_rng_state_all() (in module tensorplay.cuda.random) set_state() (tensorplay.Generator method) set_stream() (in module tensorplay.cuda) set_submodule() (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) set_sync_debug_mode() (in module tensorplay.cuda) setdefault() (tensorplay.autograd.profiler_util.StringTable method) setup_context() (tensorplay.autograd.function.Function static method) (tensorplay.autograd.function.InplaceFunction static method) (tensorplay.autograd.function.NestedIOFunction static method) sgn() (in module tensorplay.functional) sgn_() (in module tensorplay) share_memory() (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) shifted_chebyshev_polynomial_t() (in module tensorplay) (in module tensorplay.functional) shifted_chebyshev_polynomial_u() (in module tensorplay) (in module tensorplay.functional) shifted_chebyshev_polynomial_v() (in module tensorplay) (in module tensorplay.functional) shifted_chebyshev_polynomial_w() (in module tensorplay) (in module tensorplay.functional) sigmoid() (in module tensorplay.functional) (in module tensorplay.nn.functional) sigmoid_() (in module tensorplay.functional) sign() (in module tensorplay.functional) sign_() (in module tensorplay) signbit() (in module tensorplay.functional) silu() (in module tensorplay) (in module tensorplay.nn.functional) silu_() (in module tensorplay) silu_and_mul() (in module tensorplay) silu_mul() (in module tensorplay) sin() (in module tensorplay.functional) (tensorplay.SymFloat method) sin_() (in module tensorplay.functional) sinc() (in module tensorplay.functional) sinc_() (in module tensorplay.functional) sinh() (in module tensorplay.functional) (tensorplay.SymFloat method) sinh_() (in module tensorplay.functional) Size (class in tensorplay) size() (in module tensorplay) (tensorplay.UntypedStorage method) skip_data (class in tensorplay.serialization) slice() (in module tensorplay) slice_copy() (in module tensorplay.functional) slice_inverse() (in module tensorplay.functional) slice_scatter() (in module tensorplay.functional) slogdet() (in module tensorplay.functional) slow_conv3d() (in module tensorplay) slow_conv3d_forward() (in module tensorplay) slow_conv_dilated2d() (in module tensorplay) slow_conv_dilated3d() (in module tensorplay) slow_conv_transpose2d() (in module tensorplay) slow_conv_transpose3d() (in module tensorplay) smm() (in module tensorplay) smooth_l1_loss() (in module tensorplay) (in module tensorplay.nn.functional) soft_margin_loss() (in module tensorplay) (in module tensorplay.nn.functional) softmax() (in module tensorplay.functional) (in module tensorplay.nn.functional) (in module tensorplay.special._elementwise) softmin() (in module tensorplay.nn.functional) softplus() (in module tensorplay) (in module tensorplay.nn.functional) softshrink() (in module tensorplay) (in module tensorplay.nn.functional) softsign() (in module tensorplay.nn.functional) sort() (in module tensorplay.functional) (tensorplay.autograd.profiler_util.EventList method) sorted_indices (tensorplay.nn.utils.rnn.PackedSequence attribute) sparse_() (in module tensorplay.nn.init) sparse_add() (in module tensorplay) sparse_bsc_tensor() (in module tensorplay.functional) sparse_bsr_tensor() (in module tensorplay.functional) sparse_compressed_tensor() (in module tensorplay) sparse_coo_tensor() (in module tensorplay.functional) sparse_csc_tensor() (in module tensorplay.functional) sparse_csr_tensor() (in module tensorplay.functional) sparse_dim() (in module tensorplay) sparse_mask() (in module tensorplay) sparse_mm() (in module tensorplay) sparse_mul() (in module tensorplay) sparse_resize_() (in module tensorplay) sparse_resize_and_clear_() (in module tensorplay) sparse_sampled_addmm() (in module tensorplay) sparse_sum() (in module tensorplay) spdiags() (in module tensorplay) special_airy_ai() (in module tensorplay) special_bessel_j0() (in module tensorplay) special_bessel_j1() (in module tensorplay) special_bessel_y0() (in module tensorplay) special_bessel_y1() (in module tensorplay) special_chebyshev_polynomial_t() (in module tensorplay) special_chebyshev_polynomial_u() (in module tensorplay) special_chebyshev_polynomial_v() (in module tensorplay) special_chebyshev_polynomial_w() (in module tensorplay) special_digamma() (in module tensorplay) special_entr() (in module tensorplay) special_erf() (in module tensorplay) special_erfc() (in module tensorplay) special_erfcx() (in module tensorplay) special_erfinv() (in module tensorplay) special_exp2() (in module tensorplay) special_expit() (in module tensorplay) special_expm1() (in module tensorplay) special_gammainc() (in module tensorplay) special_gammaincc() (in module tensorplay) special_gammaln() (in module tensorplay) special_hermite_polynomial_h() (in module tensorplay) special_hermite_polynomial_he() (in module tensorplay) special_i0() (in module tensorplay) special_i0e() (in module tensorplay) special_i1() (in module tensorplay) special_i1e() (in module tensorplay) special_laguerre_polynomial_l() (in module tensorplay) special_legendre_polynomial_p() (in module tensorplay) special_log1p() (in module tensorplay) special_log_ndtr() (in module tensorplay) special_log_softmax() (in module tensorplay) special_logit() (in module tensorplay) special_logsumexp() (in module tensorplay) special_modified_bessel_i0() (in module tensorplay) special_modified_bessel_i1() (in module tensorplay) special_modified_bessel_k0() (in module tensorplay) special_modified_bessel_k1() (in module tensorplay) special_multigammaln() (in module tensorplay) special_ndtr() (in module tensorplay) special_ndtri() (in module tensorplay) special_polygamma() (in module tensorplay) special_psi() (in module tensorplay) special_round() (in module tensorplay) special_scaled_modified_bessel_k0() (in module tensorplay) special_scaled_modified_bessel_k1() (in module tensorplay) special_shifted_chebyshev_polynomial_t() (in module tensorplay) special_shifted_chebyshev_polynomial_u() (in module tensorplay) special_shifted_chebyshev_polynomial_v() (in module tensorplay) special_shifted_chebyshev_polynomial_w() (in module tensorplay) special_sinc() (in module tensorplay) special_softmax() (in module tensorplay) special_spherical_bessel_j0() (in module tensorplay) special_xlog1py() (in module tensorplay) special_xlogy() (in module tensorplay) special_zeta() (in module tensorplay) spherical_bessel_j0() (in module tensorplay) (in module tensorplay.functional) split() (in module tensorplay.functional) split_copy() (in module tensorplay.functional) split_with_sizes() (in module tensorplay) split_with_sizes_copy() (in module tensorplay.functional) sqrt() (in module tensorplay.functional) (tensorplay.SymFloat method) sqrt_() (in module tensorplay.functional) square() (in module tensorplay.functional) square_() (in module tensorplay.functional) squeeze() (in module tensorplay.functional) squeeze_() (in module tensorplay) squeeze_copy() (in module tensorplay.functional) sspaddmm() (in module tensorplay) stack() (in module tensorplay.functional) StackDataset (class in tensorplay.utils.data) stage_and_launch() (tensorplay.cuda.graphs.CUDAGraph method) start() (in module tensorplay.cuda.profiler) state_dict() (tensorplay.GradScaler method) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) statically_known_true() (tensorplay.SymBool method) stax() (in module tensorplay.stax) std() (in module tensorplay.functional) std_mean() (in module tensorplay.functional) step() (tensorplay.GradScaler method) (tensorplay.profiler.profiler.profile method) stft() (in module tensorplay.functional) stop() (in module tensorplay.cuda.profiler) storage_offset() (in module tensorplay) StorageType (in module tensorplay.serialization) Stream (class in tensorplay.cuda.streams) stream() (in module tensorplay.cuda) StreamContext (class in tensorplay.cuda) stride() (in module tensorplay) StringTable (class in tensorplay.autograd.profiler_util) sub() (in module tensorplay.functional) sub_() (in module tensorplay) Subset (class in tensorplay.utils.data) SubsetRandomSampler (class in tensorplay.utils.data) subtract() (in module tensorplay.functional) subtract_() (in module tensorplay) sum() (in module tensorplay.functional) sum_to_size() (in module tensorplay) svd() (in module tensorplay.functional) swapaxes() (in module tensorplay.functional) swapaxes_() (in module tensorplay) swapdims() (in module tensorplay.functional) swapdims_() (in module tensorplay) sym_and() (tensorplay.SymBool method) sym_constrain_range() (in module tensorplay.functional) sym_constrain_range_for_size() (in module tensorplay.functional) sym_eq() (tensorplay.SymFloat method) (tensorplay.SymInt method) sym_float() (in module tensorplay) (tensorplay.SymFloat method) (tensorplay.SymInt method) sym_ge() (tensorplay.SymFloat method) (tensorplay.SymInt method) sym_gt() (tensorplay.SymFloat method) (tensorplay.SymInt method) sym_int() (in module tensorplay) (tensorplay.SymFloat method) sym_is_contiguous() (in module tensorplay) sym_ite() (in module tensorplay) sym_le() (tensorplay.SymFloat method) (tensorplay.SymInt method) sym_lt() (tensorplay.SymFloat method) (tensorplay.SymInt method) sym_max() (in module tensorplay) sym_min() (in module tensorplay) sym_ne() (tensorplay.SymFloat method) (tensorplay.SymInt method) sym_not() (in module tensorplay) (tensorplay.SymBool method) sym_numel() (in module tensorplay) sym_or() (tensorplay.SymBool method) sym_size() (in module tensorplay) sym_sqrt() (in module tensorplay) sym_storage_offset() (in module tensorplay) sym_stride() (in module tensorplay) sym_sum() (in module tensorplay) sym_xor() (tensorplay.SymBool method) symbolic() (tensorplay.SymBool static method) (tensorplay.SymFloat static method) (tensorplay.SymInt static method) SymBool (class in tensorplay) SymFloat (class in tensorplay) SymInt (class in tensorplay) synchronize() (in module tensorplay.cuda) (tensorplay.cuda.streams.Event method) (tensorplay.cuda.streams.ExternalStream method) (tensorplay.cuda.streams.Stream method) T t() (in module tensorplay.functional) t_() (in module tensorplay) t_copy() (in module tensorplay.functional) table() (tensorplay.autograd.profiler_util.EventList method) take() (in module tensorplay.functional) take_along_dim() (in module tensorplay.functional) tan() (in module tensorplay.functional) (tensorplay.SymFloat method) tan_() (in module tensorplay.functional) tanh() (in module tensorplay.functional) (in module tensorplay.nn.functional) (tensorplay.SymFloat method) tanh_() (in module tensorplay.functional) tanhshrink() (in module tensorplay.nn.functional) temperature() (in module tensorplay.cuda) Tensor (in module tensorplay) tensor() (in module tensorplay) tensor_split() (in module tensorplay.functional) TensorDataset (class in tensorplay.utils.data) tensordot() (in module tensorplay._shape_funcs) tensorplay.amp.autocast_mode module tensorplay.amp.grad_scaler module tensorplay.cpu.amp module tensorplay.cpu.amp.autocast_mode module tensorplay.cpu.amp.grad_scaler module tensorplay.cuda.amp module tensorplay.cuda.amp.autocast_mode module tensorplay.cuda.amp.common module tensorplay.cuda.amp.grad_scaler module tensorplay.distributed.algorithms module tensorplay.distributed.algorithms.ddp_comm_hooks module tensorplay.distributed.algorithms.ddp_comm_hooks.ddp_zero_hook module tensorplay.distributed.algorithms.ddp_comm_hooks.debugging_hooks module tensorplay.distributed.algorithms.ddp_comm_hooks.default_hooks module tensorplay.distributed.algorithms.ddp_comm_hooks.mixed_precision_hooks module tensorplay.distributed.algorithms.ddp_comm_hooks.optimizer_overlap_hooks module tensorplay.distributed.algorithms.ddp_comm_hooks.post_localSGD_hook module tensorplay.distributed.algorithms.ddp_comm_hooks.powerSGD_hook module tensorplay.distributed.algorithms.ddp_comm_hooks.quantization_hooks module tensorplay.distributed.algorithms.join module tensorplay.distributed.algorithms.model_averaging module tensorplay.distributed.algorithms.model_averaging.averagers module tensorplay.distributed.algorithms.model_averaging.hierarchical_model_averager module tensorplay.distributed.algorithms.model_averaging.utils module tensorplay.distributed.argparse_util module tensorplay.distributed.checkpoint.api module tensorplay.distributed.checkpoint.default_planner module tensorplay.distributed.checkpoint.filesystem module tensorplay.distributed.checkpoint.hf_storage module tensorplay.distributed.checkpoint.metadata module tensorplay.distributed.checkpoint.optimizer module tensorplay.distributed.checkpoint.planner module tensorplay.distributed.checkpoint.planner_helpers module tensorplay.distributed.checkpoint.quantized_hf_storage module tensorplay.distributed.checkpoint.resharding module tensorplay.distributed.checkpoint.state_dict module tensorplay.distributed.checkpoint.state_dict_loader module tensorplay.distributed.checkpoint.state_dict_saver module tensorplay.distributed.checkpoint.stateful module tensorplay.distributed.checkpoint.storage module tensorplay.distributed.checkpoint.utils module tensorplay.distributed.constants module tensorplay.distributed.device_mesh module tensorplay.distributed.distributed_core module tensorplay.distributed.elastic module tensorplay.distributed.elastic.agent.server.api module tensorplay.distributed.elastic.agent.server.local_elastic_agent module tensorplay.distributed.elastic.events.api module tensorplay.distributed.elastic.events.handlers module tensorplay.distributed.elastic.metrics.api module tensorplay.distributed.elastic.multiprocessing.api module tensorplay.distributed.elastic.multiprocessing.errors.error_handler module tensorplay.distributed.elastic.multiprocessing.errors.handlers module tensorplay.distributed.elastic.multiprocessing.redirects module tensorplay.distributed.elastic.multiprocessing.tail_log module tensorplay.distributed.elastic.rendezvous.api module tensorplay.distributed.elastic.rendezvous.c10d_rendezvous_backend module tensorplay.distributed.elastic.rendezvous.dynamic_rendezvous module tensorplay.distributed.elastic.rendezvous.etcd_rendezvous module tensorplay.distributed.elastic.rendezvous.etcd_rendezvous_backend module tensorplay.distributed.elastic.rendezvous.etcd_server module tensorplay.distributed.elastic.rendezvous.etcd_store module tensorplay.distributed.elastic.rendezvous.static_tcp_rendezvous module tensorplay.distributed.elastic.rendezvous.utils module tensorplay.distributed.elastic.timer.api module tensorplay.distributed.elastic.timer.file_based_local_timer module tensorplay.distributed.elastic.timer.local_timer module tensorplay.distributed.elastic.utils module tensorplay.distributed.elastic.utils.api module tensorplay.distributed.elastic.utils.data module tensorplay.distributed.elastic.utils.data.cycling_iterator module tensorplay.distributed.elastic.utils.data.elastic_distributed_sampler module tensorplay.distributed.elastic.utils.distributed module tensorplay.distributed.elastic.utils.log_level module tensorplay.distributed.elastic.utils.logging module tensorplay.distributed.elastic.utils.store module tensorplay.distributed.fsdp.api module tensorplay.distributed.fsdp.fully_sharded_data_parallel module tensorplay.distributed.fsdp.sharded_grad_scaler module tensorplay.distributed.fsdp.wrap module tensorplay.distributed.launcher module tensorplay.distributed.logging_handlers module tensorplay.distributed.nn module tensorplay.distributed.nn.api module tensorplay.distributed.nn.api.remote_module module tensorplay.distributed.nn.functional module tensorplay.distributed.nn.jit module tensorplay.distributed.nn.jit.instantiator module tensorplay.distributed.nn.jit.templates module tensorplay.distributed.nn.jit.templates.remote_module_template module tensorplay.distributed.optim.apply_optimizer_in_backward module tensorplay.distributed.optim.functional_adadelta module tensorplay.distributed.optim.functional_adagrad module tensorplay.distributed.optim.functional_adam module tensorplay.distributed.optim.functional_adamax module tensorplay.distributed.optim.functional_adamw module tensorplay.distributed.optim.functional_rmsprop module tensorplay.distributed.optim.functional_rprop module tensorplay.distributed.optim.functional_sgd module tensorplay.distributed.optim.named_optimizer module tensorplay.distributed.optim.optimizer module tensorplay.distributed.optim.post_localSGD_optimizer module tensorplay.distributed.optim.utils module tensorplay.distributed.optim.zero_redundancy_optimizer module tensorplay.distributed.p10d_logger module tensorplay.distributed.remote_device module tensorplay.distributed.rendezvous module tensorplay.distributed.rpc.api module tensorplay.distributed.rpc.backend_registry module tensorplay.distributed.rpc.constants module tensorplay.distributed.rpc.functions module tensorplay.distributed.rpc.internal module tensorplay.distributed.rpc.options module tensorplay.distributed.rpc.rref_proxy module tensorplay.distributed.rpc.server_process_global_profiler module tensorplay.distributed.tensor.parallel.api module tensorplay.distributed.tensor.parallel.ddp module tensorplay.distributed.tensor.parallel.fsdp module tensorplay.distributed.tensor.parallel.input_reshard module tensorplay.distributed.tensor.parallel.loss module tensorplay.distributed.tensor.parallel.style module tensorplay.distributed.utils module tensorplay.multiprocessing.pool module tensorplay.multiprocessing.queue module tensorplay.multiprocessing.reductions module tensorplay.nn.backends module tensorplay.nn.backends.thnn module tensorplay.nn.common_types module tensorplay.nn.cpp module tensorplay.nn.functional module tensorplay.nn.grad module tensorplay.nn.init module tensorplay.nn.modules.activation module tensorplay.nn.modules.adaptive module tensorplay.nn.modules.batchnorm module tensorplay.nn.modules.channelshuffle module tensorplay.nn.modules.container module tensorplay.nn.modules.conv module tensorplay.nn.modules.distance module tensorplay.nn.modules.dropout module tensorplay.nn.modules.flatten module tensorplay.nn.modules.folding module tensorplay.nn.modules.instancenorm module tensorplay.nn.modules.lazy module tensorplay.nn.modules.linear module tensorplay.nn.modules.loss module tensorplay.nn.modules.module module tensorplay.nn.modules.normalization module tensorplay.nn.modules.padding module tensorplay.nn.modules.pixelshuffle module tensorplay.nn.modules.pooling module tensorplay.nn.modules.rnn module tensorplay.nn.modules.sparse module tensorplay.nn.modules.transformer module tensorplay.nn.modules.upsampling module tensorplay.nn.modules.utils module tensorplay.nn.parallel.comm module tensorplay.nn.parallel.distributed module tensorplay.nn.parallel.parallel_apply module tensorplay.nn.parallel.replicate module tensorplay.nn.parallel.scatter_gather module tensorplay.nn.parameter module tensorplay.nn.utils.clip_grad module tensorplay.nn.utils.convert_parameters module tensorplay.nn.utils.fusion module tensorplay.nn.utils.init module tensorplay.nn.utils.memory_format module tensorplay.nn.utils.parametrizations module tensorplay.nn.utils.parametrize module tensorplay.nn.utils.prune module tensorplay.nn.utils.rnn module tensorplay.nn.utils.stateless module tensorplay.optim.lr_scheduler module tensorplay.optim.optimizer module tensorplay.optim.swa_utils module tensorplay.utils.data.datapipes module tensorplay.utils.data.datapipes.dataframe module tensorplay.utils.data.datapipes.iter module tensorplay.utils.data.datapipes.map module tensorplay.utils.data.datapipes.utils module tensorplay.utils.serialization module tensorplay.utils.serialization.config module then() (tensorplay.Future method) (tensorplay.futures.Future method) thnn_conv2d() (in module tensorplay) threshold() (in module tensorplay.functional) (in module tensorplay.nn.functional) threshold_() (in module tensorplay.functional) (in module tensorplay.nn.functional) tile() (in module tensorplay.functional) to() (in module tensorplay) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) (tensorplay.nn.utils.rnn.PackedSequence method) to_bytes() (tensorplay.Layout method) (tensorplay.MemoryFormat method) (tensorplay.QScheme method) to_dense() (in module tensorplay) to_empty() (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) to_mkldnn() (in module tensorplay) to_padded_tensor() (in module tensorplay) to_sparse() (in module tensorplay) to_sparse_bsc() (in module tensorplay) to_sparse_bsr() (in module tensorplay) to_sparse_csc() (in module tensorplay) to_sparse_csr() (in module tensorplay) to_sym_float() (tensorplay.SymBool method) (tensorplay.SymInt method) to_sym_int() (tensorplay.SymBool method) topk() (in module tensorplay.functional) total_average() (tensorplay.autograd.profiler_util.EventList method) tp_cosine_embedding_loss() (in module tensorplay) tp_hinge_embedding_loss() (in module tensorplay) tp_kl_div() (in module tensorplay) tp_l1_loss() (in module tensorplay) tp_margin_ranking_loss() (in module tensorplay) tp_poisson_nll_loss() (in module tensorplay) tp_soft_margin_loss() (in module tensorplay) trace() (in module tensorplay.functional) train() (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) transpose() (in module tensorplay.functional) transpose_() (in module tensorplay) transpose_copy() (in module tensorplay.functional) trapezoid() (in module tensorplay.functional) trapz() (in module tensorplay.functional) triangular_solve() (in module tensorplay.functional) tril() (in module tensorplay.functional) tril_() (in module tensorplay) tril_indices() (in module tensorplay.functional) triplet_margin_loss() (in module tensorplay.functional) (in module tensorplay.nn.functional) triplet_margin_with_distance_loss() (in module tensorplay.nn.functional) triton_op() (in module tensorplay.library) triu() (in module tensorplay.functional) triu_() (in module tensorplay) triu_indices() (in module tensorplay.functional) true_divide() (in module tensorplay.functional) true_divide_() (in module tensorplay) trunc() (in module tensorplay.functional) (tensorplay.SymFloat method) (tensorplay.SymInt method) trunc_() (in module tensorplay.functional) trunc_normal_() (in module tensorplay.nn.init) type() (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) type_as() (in module tensorplay) typename() (in module tensorplay) U unbind() (in module tensorplay.functional) unbind_copy() (in module tensorplay.functional) Unflatten (class in tensorplay.nn.modules.flatten) unflatten() (in module tensorplay.functional) unflatten_dense_tensors() (in module tensorplay) unfold() (in module tensorplay) (in module tensorplay.nn.functional) unfold_copy() (in module tensorplay.functional) uniform_() (in module tensorplay) (in module tensorplay.nn.init) unique() (in module tensorplay.functional) unique_consecutive() (in module tensorplay.functional) unique_dim() (in module tensorplay) unique_dim_consecutive() (in module tensorplay) unpack_dual() (in module tensorplay.autograd.forward_ad) unpack_sequence() (in module tensorplay.nn.utils.rnn) unpad_sequence() (in module tensorplay.nn.utils.rnn) unravel_index() (in module tensorplay._shape_funcs) unsafe_chunk() (in module tensorplay) unsafe_split() (in module tensorplay) unsafe_split_with_sizes() (in module tensorplay) unscale_() (tensorplay.GradScaler method) unsorted_indices (tensorplay.nn.utils.rnn.PackedSequence attribute) unsqueeze() (in module tensorplay.functional) unsqueeze_() (in module tensorplay) unsqueeze_copy() (in module tensorplay.functional) UntypedStorage (class in tensorplay) update() (tensorplay.autograd.profiler_util.StringTable method) (tensorplay.GradScaler method) upsample_bicubic2d() (in module tensorplay) upsample_bilinear2d() (in module tensorplay) upsample_linear1d() (in module tensorplay) upsample_nearest1d() (in module tensorplay) upsample_nearest2d() (in module tensorplay) upsample_nearest3d() (in module tensorplay) upsample_trilinear3d() (in module tensorplay) use_deterministic_algorithms() (in module tensorplay) use_mem_pool() (in module tensorplay.cuda.memory) utilization() (in module tensorplay.cuda) V value() (tensorplay.Future method) (tensorplay.futures.Future method) values() (in module tensorplay) (tensorplay.autograd.profiler_util.StringTable method) values_copy() (in module tensorplay.functional) vander() (in module tensorplay.functional) var() (in module tensorplay.functional) var_mean() (in module tensorplay.functional) Variable (class in tensorplay.autograd.variable) VariableMeta (class in tensorplay.autograd.variable) vdot() (in module tensorplay.functional) version() (in module tensorplay.cuda.nccl) vhp() (in module tensorplay.autograd.functional) view() (in module tensorplay) view_as() (in module tensorplay) view_as_complex() (in module tensorplay.functional) view_as_complex_copy() (in module tensorplay.functional) view_as_real() (in module tensorplay.functional) view_as_real_copy() (in module tensorplay.functional) view_copy() (in module tensorplay.functional) vision_to_tensor() (in module tensorplay) vjp() (in module tensorplay.autograd.functional) vmap() (tensorplay.autograd.function.Function static method) (tensorplay.autograd.function.InplaceFunction static method) (tensorplay.autograd.function.NestedIOFunction static method) vsplit() (in module tensorplay.functional) vstack() (in module tensorplay.functional) W wait() (tensorplay.cuda.streams.Event method) (tensorplay.Future method) (tensorplay.futures.Future method) wait_all() (in module tensorplay.futures) wait_event() (tensorplay.cuda.streams.ExternalStream method) (tensorplay.cuda.streams.Stream method) wait_stream() (tensorplay.cuda.streams.ExternalStream method) (tensorplay.cuda.streams.Stream method) WeightedRandomSampler (class in tensorplay.utils.data) where() (in module tensorplay.functional) wrap_triton() (in module tensorplay.library) X xavier_normal_() (in module tensorplay.nn.init) xavier_uniform_() (in module tensorplay.nn.init) xlog1py() (in module tensorplay) (in module tensorplay.functional) xlog1py_() (in module tensorplay) xlogy() (in module tensorplay.functional) xlogy_() (in module tensorplay.functional) Z zero_() (in module tensorplay.functional) zero_grad() (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) zeros() (in module tensorplay) zeros_() (in module tensorplay.nn.init) zeros_like() (in module tensorplay.functional) zeta() (in module tensorplay) (in module tensorplay.functional)