Index _ | A | B | C | D | E | F | G | H | I | K | L | M | N | O | P | R | S | T | U | V | W | X | Z _ __call__() (oneflow.nn.Graph method) __init__() (oneflow.autograd.enable_grad method) (oneflow.autograd.inference_mode method) (oneflow.autograd.no_grad method) (oneflow.autograd.set_grad_enabled method) (oneflow.cuda.BoolTensor method) (oneflow.cuda.ByteTensor method) (oneflow.cuda.CharTensor method) (oneflow.cuda.DoubleTensor method) (oneflow.cuda.FloatTensor method) (oneflow.cuda.HalfTensor method) (oneflow.cuda.IntTensor method) (oneflow.cuda.LongTensor method) (oneflow.enable_grad method) (oneflow.inference_mode method) (oneflow.nn.AdaptiveAvgPool1d method) (oneflow.nn.AdaptiveAvgPool2d method) (oneflow.nn.AdaptiveAvgPool3d method) (oneflow.nn.AvgPool1d method) (oneflow.nn.AvgPool2d method) (oneflow.nn.AvgPool3d method) (oneflow.nn.BatchNorm1d method) (oneflow.nn.BatchNorm2d method) (oneflow.nn.BatchNorm3d method) (oneflow.nn.BCELoss method) (oneflow.nn.BCEWithLogitsLoss method) (oneflow.nn.CombinedMarginLoss method) (oneflow.nn.ConstantPad1d method) (oneflow.nn.ConstantPad2d method) (oneflow.nn.ConstantPad3d method) (oneflow.nn.Conv1d method) (oneflow.nn.Conv2d method) (oneflow.nn.Conv3d method) (oneflow.nn.ConvTranspose1d method) (oneflow.nn.ConvTranspose2d method) (oneflow.nn.ConvTranspose3d method) (oneflow.nn.CosineSimilarity method) (oneflow.nn.CrossEntropyLoss method) (oneflow.nn.CTCLoss method) (oneflow.nn.Dropout method) (oneflow.nn.Embedding method) (oneflow.nn.Flatten method) (oneflow.nn.Fold method) (oneflow.nn.FusedBatchNorm1d method) (oneflow.nn.FusedBatchNorm2d method) (oneflow.nn.FusedBatchNorm3d method) (oneflow.nn.Graph method) (oneflow.nn.GroupNorm method) (oneflow.nn.GRU method) (oneflow.nn.GRUCell method) (oneflow.nn.Identity method) (oneflow.nn.InstanceNorm1d method) (oneflow.nn.InstanceNorm2d method) (oneflow.nn.InstanceNorm3d method) (oneflow.nn.KLDivLoss method) (oneflow.nn.L1Loss method) (oneflow.nn.LayerNorm method) (oneflow.nn.Linear method) (oneflow.nn.LogSoftmax method) (oneflow.nn.LSTM method) (oneflow.nn.LSTMCell method) (oneflow.nn.MarginRankingLoss method) (oneflow.nn.MaxPool1d method) (oneflow.nn.MaxPool2d method) (oneflow.nn.MaxPool3d method) (oneflow.nn.Module method) (oneflow.nn.ModuleDict method) (oneflow.nn.ModuleList method) (oneflow.nn.MSELoss method) (oneflow.nn.NLLLoss method) (oneflow.nn.ParameterDict method) (oneflow.nn.ParameterList method) (oneflow.nn.ReflectionPad2d method) (oneflow.nn.ReplicationPad2d method) (oneflow.nn.RNN method) (oneflow.nn.RNNCell method) (oneflow.nn.Sequential method) (oneflow.nn.SmoothL1Loss method) (oneflow.nn.Softmax method) (oneflow.nn.TripletMarginLoss method) (oneflow.nn.Unfold method) (oneflow.nn.Upsample method) (oneflow.nn.UpsamplingBilinear2d method) (oneflow.nn.UpsamplingNearest2d method) (oneflow.nn.utils.rnn.PackedSequence method) (oneflow.nn.ZeroPad2d method) (oneflow.no_grad method) (oneflow.optim.Adagrad method) (oneflow.optim.Adam method) (oneflow.optim.AdamW method) (oneflow.optim.LAMB method) (oneflow.optim.lr_scheduler.ChainedScheduler method) (oneflow.optim.lr_scheduler.ConstantLR method) (oneflow.optim.lr_scheduler.CosineAnnealingLR method) (oneflow.optim.lr_scheduler.CosineAnnealingWarmRestarts method) (oneflow.optim.lr_scheduler.CosineDecayLR method) (oneflow.optim.lr_scheduler.ExponentialLR method) (oneflow.optim.lr_scheduler.LambdaLR method) (oneflow.optim.lr_scheduler.LinearLR method) (oneflow.optim.lr_scheduler.MultiStepLR method) (oneflow.optim.lr_scheduler.PolynomialLR method) (oneflow.optim.lr_scheduler.ReduceLROnPlateau method) (oneflow.optim.lr_scheduler.SequentialLR method) (oneflow.optim.lr_scheduler.StepLR method) (oneflow.optim.RMSprop method) (oneflow.optim.SGD method) (oneflow.set_grad_enabled method) __repr__() (oneflow.nn.Graph method) A abs() (in module oneflow) (oneflow.Tensor method) acos() (in module oneflow) (oneflow.Tensor method) acosh() (in module oneflow) (oneflow.Tensor method) activation_checkpointing() (oneflow.nn.graph.block_config.BlockConfig property) Adagrad (class in oneflow.optim) Adam (class in oneflow.optim) AdamW (class in oneflow.optim) adaptive_avg_pool1d() (in module oneflow.nn.functional) adaptive_avg_pool2d() (in module oneflow.nn.functional) adaptive_avg_pool3d() (in module oneflow.nn.functional) AdaptiveAvgPool1d (class in oneflow.nn) AdaptiveAvgPool2d (class in oneflow.nn) AdaptiveAvgPool3d (class in oneflow.nn) add() (in module oneflow) (oneflow.Tensor method) add_() (oneflow.Tensor method) add_optimizer() (oneflow.nn.Graph method) add_param_group() (oneflow.optim.Optimizer method) addcmul() (in module oneflow) (oneflow.Tensor method) addcmul_() (oneflow.Tensor method) addmm() (in module oneflow) (oneflow.Tensor method) affine_grid() (in module oneflow.nn.functional) all() (in module oneflow) (oneflow.Tensor method) all_device_placement() (in module oneflow.env) all_gather() (in module oneflow.comm) all_reduce() (in module oneflow.comm) all_to_all() (in module oneflow.comm) allow_fuse_add_to_output() (oneflow.nn.graph.graph_config.GraphConfig method) allow_fuse_cast_scale() (oneflow.nn.graph.graph_config.GraphConfig method) allow_fuse_model_update_ops() (oneflow.nn.graph.graph_config.GraphConfig method) amax() (in module oneflow) (oneflow.Tensor method) amin() (in module oneflow) (oneflow.Tensor method) any() (in module oneflow) (oneflow.Tensor method) arange() (in module oneflow) arccos() (in module oneflow) (oneflow.Tensor method) arccosh() (in module oneflow) (oneflow.Tensor method) arcsin() (in module oneflow) (oneflow.Tensor method) arcsinh() (in module oneflow) (oneflow.Tensor method) arctan() (in module oneflow) (oneflow.Tensor method) arctanh() (in module oneflow) (oneflow.Tensor method) argmax() (in module oneflow) (oneflow.Tensor method) argmin() (in module oneflow) (oneflow.Tensor method) argsort() (in module oneflow) (oneflow.Tensor method) argwhere() (in module oneflow) (oneflow.Tensor method) as_strided() (in module oneflow) as_tensor() (in module oneflow) asin() (in module oneflow) (oneflow.Tensor method) asinh() (in module oneflow) (oneflow.Tensor method) atan() (in module oneflow) (oneflow.Tensor method) atan2() (in module oneflow) (oneflow.Tensor method) atanh() (in module oneflow) (oneflow.Tensor method) avg_pool1d() (in module oneflow.nn.functional) avg_pool2d() (in module oneflow.nn.functional) avg_pool3d() (in module oneflow.nn.functional) AvgPool1d (class in oneflow.nn) AvgPool2d (class in oneflow.nn) AvgPool3d (class in oneflow.nn) B backward() (in module oneflow.autograd) (oneflow.Tensor method) barrier() (in module oneflow.comm) batch_align (in module oneflow.nn.image) batch_sizes (oneflow.nn.utils.rnn.PackedSequence attribute) BatchNorm1d (class in oneflow.nn) BatchNorm2d (class in oneflow.nn) BatchNorm3d (class in oneflow.nn) BatchSampler (class in oneflow.utils.data) BCELoss (class in oneflow.nn) BCEWithLogitsLoss (class in oneflow.nn) bernoulli() (in module oneflow) bias (oneflow.nn.Conv1d attribute) (oneflow.nn.Conv3d attribute) (oneflow.nn.ConvTranspose1d attribute) (oneflow.nn.ConvTranspose2d attribute) (oneflow.nn.ConvTranspose3d attribute) bias_hh (oneflow.nn.GRUCell attribute) (oneflow.nn.LSTMCell attribute) (oneflow.nn.RNNCell attribute) bias_ih (oneflow.nn.GRUCell attribute) (oneflow.nn.LSTMCell attribute) (oneflow.nn.RNNCell attribute) bmm() (in module oneflow) (oneflow.Tensor method) BoolTensor (class in oneflow.cuda) broadcast() (in module oneflow.comm) broadcast_like() (in module oneflow) build() (oneflow.nn.Graph method) byte() (oneflow.Tensor method) ByteTensor (class in oneflow.cuda) C calculate_gain() (in module oneflow.nn.init) cast() (oneflow.Tensor method) cat() (in module oneflow) ceil() (in module oneflow) (oneflow.Tensor method) CELU (class in oneflow.nn) celu() (in module oneflow.nn.functional) ChainedScheduler (class in oneflow.optim.lr_scheduler) CharTensor (class in oneflow.cuda) chunk() (in module oneflow) (oneflow.Tensor method) clamp() (in module oneflow) (oneflow.Tensor method) clamp_() (oneflow.Tensor method) clip() (in module oneflow) (oneflow.Tensor method) clip_() (oneflow.Tensor method) clip_grad_norm_() (in module oneflow.nn.utils) clip_grad_value_() (in module oneflow.nn.utils) clone() (oneflow.Tensor method) CombinedMarginLoss (class in oneflow.nn) concat() (in module oneflow) ConcatDataset (class in oneflow.utils.data) constant_() (in module oneflow.nn.init) ConstantLR (class in oneflow.optim.lr_scheduler) ConstantPad1d (class in oneflow.nn) ConstantPad2d (class in oneflow.nn) ConstantPad3d (class in oneflow.nn) contiguous() (oneflow.Tensor method) Conv1d (class in oneflow.nn) conv1d() (in module oneflow.nn.functional) Conv2d (class in oneflow.nn) conv2d() (in module oneflow.nn.functional) Conv3d (class in oneflow.nn) conv3d() (in module oneflow.nn.functional) conv_transpose1d() (in module oneflow.nn.functional) conv_transpose2d() (in module oneflow.nn.functional) conv_transpose3d() (in module oneflow.nn.functional) ConvTranspose1d (class in oneflow.nn) ConvTranspose2d (class in oneflow.nn) ConvTranspose3d (class in oneflow.nn) copy_() (oneflow.Tensor method) cos() (in module oneflow) (oneflow.Tensor method) cosh() (in module oneflow) (oneflow.Tensor method) cosine_similarity() (in module oneflow.nn.functional) CosineAnnealingLR (class in oneflow.optim.lr_scheduler) CosineAnnealingWarmRestarts (class in oneflow.optim.lr_scheduler) CosineDecayLR (class in oneflow.optim.lr_scheduler) CosineSimilarity (class in oneflow.nn) cpu() (oneflow.Tensor method) cross_entropy() (in module oneflow.nn.functional) CrossEntropyLoss (class in oneflow.nn) CTCLoss (class in oneflow.nn) cuda() (oneflow.Tensor method) cumprod() (in module oneflow) (oneflow.Tensor method) cumsum() (in module oneflow) (oneflow.Tensor method) current_device() (in module oneflow.cuda) D data (oneflow.nn.utils.rnn.PackedSequence attribute) (oneflow.Tensor attribute) DataLoader (class in oneflow.utils.data) Dataset (class in oneflow.utils.data) debug() (oneflow.nn.Graph method) decode (in module oneflow.nn.image) detach() (oneflow.Tensor method) device (class in oneflow) (oneflow.Tensor attribute) device_count() (in module oneflow.cuda) diag() (in module oneflow) (oneflow.Tensor method) diagonal() (in module oneflow) (in module oneflow.linalg) (oneflow.Tensor method) dim() (oneflow.Tensor method) DistributedDataParallel() (in module oneflow.nn.parallel) DistributedSampler (class in oneflow.utils.data.distributed) div() (in module oneflow) (oneflow.Tensor method) div_() (oneflow.Tensor method) dot() (in module oneflow) (oneflow.Tensor method) double() (oneflow.Tensor method) DoubleTensor (class in oneflow.cuda) Dropout (class in oneflow.nn) dropout() (in module oneflow.nn.functional) dtype (class in oneflow) (oneflow.Tensor attribute) E einsum() (in module oneflow) element_size() (oneflow.Tensor method) ELU (class in oneflow.nn) elu() (in module oneflow.nn.functional) Embedding (class in oneflow.nn) embedding() (in module oneflow.nn.functional) empty() (in module oneflow) enable_amp() (oneflow.nn.graph.graph_config.GraphConfig method) enable_cudnn_conv_heuristic_search_algo() (oneflow.nn.graph.graph_config.GraphConfig method) enable_grad (class in oneflow) (class in oneflow.autograd) enable_zero() (oneflow.nn.graph.graph_config.GraphConfig method) eq() (in module oneflow) (oneflow.Tensor method) equal() (in module oneflow) erf() (in module oneflow) (oneflow.Tensor method) erfc() (in module oneflow) (oneflow.Tensor method) erfinv() (in module oneflow) (oneflow.Tensor method) erfinv_() (oneflow.Tensor method) exp() (in module oneflow) (oneflow.Tensor method) expand() (oneflow.Tensor method) expand_as() (oneflow.Tensor method) expm1() (in module oneflow) (oneflow.Tensor method) ExponentialLR (class in oneflow.optim.lr_scheduler) eye() (in module oneflow) F fill_() (oneflow.Tensor method) Flatten (class in oneflow.nn) flatten() (in module oneflow) (oneflow.Tensor method) flip (in module oneflow.nn.image) flip() (in module oneflow) (oneflow.Tensor method) float() (oneflow.Tensor method) FloatTensor (class in oneflow.cuda) floor() (in module oneflow) (oneflow.Tensor method) floor_() (in module oneflow) (oneflow.Tensor method) floor_divide() (in module oneflow) (oneflow.Tensor method) fmod() (in module oneflow) (oneflow.Tensor method) Fold (class in oneflow.nn) forward() (oneflow.one_embedding.MultiTableEmbedding method) (oneflow.one_embedding.MultiTableMultiColumnEmbedding method) from_numpy() (in module oneflow) Ftrl (class in oneflow.one_embedding) full() (in module oneflow) full_like() (in module oneflow) Function (class in oneflow.autograd) FusedBatchNorm1d (class in oneflow.nn) FusedBatchNorm2d (class in oneflow.nn) FusedBatchNorm3d (class in oneflow.nn) G gather() (in module oneflow) (in module oneflow.comm) (oneflow.Tensor method) ge() (in module oneflow) (oneflow.Tensor method) GELU (class in oneflow.nn) gelu() (in module oneflow.nn.functional) (oneflow.Tensor method) get_device() (oneflow.Tensor method) get_local_rank() (in module oneflow.env) get_node_size() (in module oneflow.env) get_rank() (in module oneflow.env) get_rng_state() (in module oneflow) get_world_size() (in module oneflow.env) global_to_global() (oneflow.Tensor method) GLU (class in oneflow.nn) glu() (in module oneflow.nn.functional) grad (oneflow.Tensor attribute) grad() (in module oneflow.autograd) grad_fn (oneflow.Tensor attribute) greater() (in module oneflow) greater_equal() (in module oneflow) grid_sample() (in module oneflow.nn.functional) GroupNorm (class in oneflow.nn) GRU (class in oneflow.nn) GRUCell (class in oneflow.nn) gt() (in module oneflow) (oneflow.Tensor method) H half() (oneflow.Tensor method) HalfTensor (class in oneflow.cuda) Hardshrink (class in oneflow.nn) hardshrink() (in module oneflow.nn.functional) Hardsigmoid (class in oneflow.nn) hardsigmoid() (in module oneflow.nn.functional) Hardswish (class in oneflow.nn) hardswish() (in module oneflow.nn.functional) Hardtanh (class in oneflow.nn) hardtanh() (in module oneflow.nn.functional) hsplit() (in module oneflow) I Identity (class in oneflow.nn) in_top_k() (oneflow.Tensor method) index_select() (in module oneflow) (oneflow.Tensor method) inference_mode (class in oneflow) (class in oneflow.autograd) init_rdma() (in module oneflow.env) initial_seed() (in module oneflow) InstanceNorm1d (class in oneflow.nn) InstanceNorm2d (class in oneflow.nn) InstanceNorm3d (class in oneflow.nn) int() (oneflow.Tensor method) interpolate() (in module oneflow.nn.functional) IntTensor (class in oneflow.cuda) is_available() (in module oneflow.cuda) is_contiguous() (oneflow.Tensor method) is_cuda (oneflow.Tensor attribute) is_floating_point() (in module oneflow) (oneflow.Tensor method) is_global (oneflow.Tensor attribute) is_grad_enabled() (in module oneflow) is_lazy (oneflow.Tensor attribute) is_leaf (oneflow.Tensor attribute) is_nonzero() (in module oneflow) is_pinned() (oneflow.Tensor method) is_tensor() (in module oneflow) isinf() (in module oneflow) (oneflow.Tensor method) isnan() (in module oneflow) (oneflow.Tensor method) item() (oneflow.Tensor method) IterableDataset (class in oneflow.utils.data) K kaiming_normal_() (in module oneflow.nn.init) kaiming_uniform_() (in module oneflow.nn.init) KLDivLoss (class in oneflow.nn) L L1Loss (class in oneflow.nn) LAMB (class in oneflow.optim) LambdaLR (class in oneflow.optim.lr_scheduler) layer_norm() (in module oneflow.nn.functional) LayerNorm (class in oneflow.nn) le() (in module oneflow) (oneflow.Tensor method) leaky_relu() (in module oneflow.nn.functional) LeakyReLU (class in oneflow.nn) Linear (class in oneflow.nn) linear() (in module oneflow.nn.functional) LinearLR (class in oneflow.optim.lr_scheduler) linspace() (in module oneflow) load() (in module oneflow) load_snapshot() (oneflow.one_embedding.MultiTableEmbedding method) (oneflow.one_embedding.MultiTableMultiColumnEmbedding method) load_state_dict() (oneflow.nn.Graph method) (oneflow.optim.Optimizer method) local_to_global() (oneflow.Tensor method) log() (in module oneflow) (oneflow.Tensor method) log1p() (in module oneflow) log2() (in module oneflow) (oneflow.Tensor method) log_softmax() (in module oneflow.nn.functional) logical_and() (in module oneflow) (oneflow.Tensor method) logical_not() (in module oneflow) (oneflow.Tensor method) logical_or() (in module oneflow) (oneflow.Tensor method) logical_xor() (in module oneflow) (oneflow.Tensor method) LogSigmoid (class in oneflow.nn) logsigmoid() (in module oneflow.nn.functional) LogSoftmax (class in oneflow.nn) long() (oneflow.Tensor method) LongTensor (class in oneflow.cuda) LSTM (class in oneflow.nn) LSTMCell (class in oneflow.nn) lt() (in module oneflow) (oneflow.Tensor method) M make_cached_host_mem_store_options() (in module oneflow.one_embedding) make_cached_ssd_store_options() (in module oneflow.one_embedding) make_device_mem_store_options() (in module oneflow.one_embedding) make_normal_initializer() (in module oneflow.one_embedding) make_persistent_table_reader() (in module oneflow.one_embedding) make_persistent_table_writer() (in module oneflow.one_embedding) make_table() (in module oneflow.one_embedding) make_table_options() (in module oneflow.one_embedding) make_uniform_initializer() (in module oneflow.one_embedding) manual_seed() (in module oneflow) (in module oneflow.cuda) manual_seed_all() (in module oneflow.cuda) MarginRankingLoss (class in oneflow.nn) mark_non_differentiable() (oneflow.autograd.autograd_function.FunctionAutoGradCaptureState method) masked_fill() (oneflow.Tensor method) masked_select() (in module oneflow) (oneflow.Tensor method) matmul() (in module oneflow) (oneflow.Tensor method) matrix_norm() (in module oneflow.linalg) max() (in module oneflow) (oneflow.Tensor method) max_pool1d() (in module oneflow.nn.functional) max_pool2d() (in module oneflow.nn.functional) max_pool3d() (in module oneflow.nn.functional) maximum() (in module oneflow) (oneflow.Tensor method) MaxPool1d (class in oneflow.nn) MaxPool2d (class in oneflow.nn) MaxPool3d (class in oneflow.nn) mean() (in module oneflow) (oneflow.Tensor method) median() (in module oneflow) (oneflow.Tensor method) meshgrid() (in module oneflow) min() (in module oneflow) (oneflow.Tensor method) minimum() (in module oneflow) (oneflow.Tensor method) Mish (class in oneflow.nn) mish() (in module oneflow.nn.functional) (oneflow.Tensor method) mm() (in module oneflow) (oneflow.Tensor method) module oneflow.one_embedding oneflow.utils.data Module (class in oneflow.nn) ModuleDict (class in oneflow.nn) ModuleList (class in oneflow.nn) movedim() (in module oneflow) MSELoss (class in oneflow.nn) mul() (in module oneflow) (oneflow.Tensor method) mul_() (oneflow.Tensor method) MultiStepLR (class in oneflow.optim.lr_scheduler) MultiTableEmbedding() (in module oneflow.one_embedding) MultiTableMultiColumnEmbedding() (in module oneflow.one_embedding) mv() (in module oneflow) (oneflow.Tensor method) N name() (oneflow.nn.Graph property) narrow() (in module oneflow) (oneflow.Tensor method) ndim() (oneflow.Tensor property) ndimension() (oneflow.Tensor method) ne() (in module oneflow) (oneflow.Tensor method) neg() (in module oneflow) (oneflow.Tensor method) negative() (in module oneflow) (oneflow.Tensor method) nelement() (oneflow.Tensor method) new_empty() (oneflow.Tensor method) new_ones() (oneflow.Tensor method) new_tensor() (oneflow.Tensor method) new_zeros() (oneflow.Tensor method) NLLLoss (class in oneflow.nn) nms() (oneflow.Tensor method) no_grad (class in oneflow) (class in oneflow.autograd) nonzero() (in module oneflow) (oneflow.Tensor method) norm() (in module oneflow) (in module oneflow.linalg) (oneflow.Tensor method) normal() (in module oneflow) normal_() (in module oneflow.nn.init) (oneflow.Tensor method) normalize (in module oneflow.nn.image) normalize() (in module oneflow.nn.functional) not_equal() (in module oneflow) numel() (in module oneflow) (oneflow.Tensor method) numpy() (oneflow.Tensor method) O one_hot() (in module oneflow.nn.functional) oneflow.one_embedding module oneflow.utils.data module ones() (in module oneflow) ones_() (in module oneflow.nn.init) ones_like() (in module oneflow) Optimizer (class in oneflow.optim) orthogonal_() (in module oneflow.nn.init) P pack_padded_sequence() (in module oneflow.nn.utils.rnn) pack_sequence() (in module oneflow.nn.utils.rnn) PackedSequence (class in oneflow.nn.utils.rnn) pad() (in module oneflow.nn.functional) pad_packed_sequence() (in module oneflow.nn.utils.rnn) pad_sequence() (in module oneflow.nn.utils.rnn) ParameterDict (class in oneflow.nn) ParameterList (class in oneflow.nn) permute() (in module oneflow) (oneflow.Tensor method) pin_memory() (oneflow.Tensor method) PixelShuffle (in module oneflow.nn) placement (class in oneflow) (oneflow.Tensor attribute) PolynomialLR (class in oneflow.optim.lr_scheduler) pow() (in module oneflow) (oneflow.Tensor method) PReLU (class in oneflow.nn) prelu() (in module oneflow.nn.functional) prod() (in module oneflow) (oneflow.Tensor method) R rand() (in module oneflow) randint() (in module oneflow) randn() (in module oneflow) random_split() (in module oneflow.utils.data) RandomSampler (class in oneflow.utils.data) randperm() (in module oneflow) rdma_is_initialized() (in module oneflow.env) reciprocal() (in module oneflow) (oneflow.Tensor method) recv() (in module oneflow.comm) reduce() (in module oneflow.comm) reduce_scatter() (in module oneflow.comm) ReduceLROnPlateau (class in oneflow.optim.lr_scheduler) ReflectionPad2d (class in oneflow.nn) register_hook() (oneflow.Tensor method) ReLU (class in oneflow.nn) relu() (in module oneflow.nn.functional) (oneflow.Tensor method) ReLU6 (class in oneflow.nn) relu6() (in module oneflow.nn.functional) remove_weight_norm() (in module oneflow.nn.utils) repeat() (oneflow.Tensor method) repeat_interleave() (in module oneflow) (oneflow.Tensor method) ReplicationPad2d (class in oneflow.nn) requires_grad (oneflow.Tensor attribute) requires_grad_() (oneflow.Tensor method) reshape() (in module oneflow) (oneflow.Tensor method) reshape_as() (oneflow.Tensor method) Resize (in module oneflow.nn.image) retain_grad() (oneflow.Tensor method) RMSprop (class in oneflow.optim) RNN (class in oneflow.nn) RNNCell (class in oneflow.nn) roll() (in module oneflow) (oneflow.Tensor method) round() (in module oneflow) (oneflow.Tensor method) rsqrt() (in module oneflow) (oneflow.Tensor method) S Sampler (class in oneflow.utils.data) save() (in module oneflow) save_for_backward() (oneflow.autograd.autograd_function.FunctionAutoGradCaptureState method) save_snapshot() (oneflow.one_embedding.MultiTableEmbedding method) (oneflow.one_embedding.MultiTableMultiColumnEmbedding method) saved_tensors() (oneflow.autograd.autograd_function.FunctionAutoGradCaptureState property) sbp (class in oneflow.sbp) (oneflow.Tensor attribute) scatter() (in module oneflow) (in module oneflow.comm) scatter_add() (in module oneflow) scatter_nd() (in module oneflow) searchsorted() (in module oneflow) seed() (in module oneflow) select() (in module oneflow) SELU (class in oneflow.nn) selu() (in module oneflow.nn.functional) (oneflow.Tensor method) send() (in module oneflow.comm) Sequential (class in oneflow.nn) SequentialLR (class in oneflow.optim.lr_scheduler) SequentialSampler (class in oneflow.utils.data) set_device() (in module oneflow.cuda) set_grad_enabled (class in oneflow) (class in oneflow.autograd) set_grad_scaler() (oneflow.nn.Graph method) set_gradient_accumulation_steps() (oneflow.nn.graph.graph_config.GraphConfig method) set_num_threads() (in module oneflow) set_printoptions() (in module oneflow) set_rng_state() (in module oneflow) set_stage() (oneflow.nn.graph.block_config.BlockConfig method) SGD (class in oneflow.optim) shape (oneflow.Tensor attribute) Sigmoid (class in oneflow.nn) sigmoid() (in module oneflow) (in module oneflow.nn.functional) (oneflow.Tensor method) sign() (in module oneflow) (oneflow.Tensor method) SiLU (class in oneflow.nn) silu() (in module oneflow.nn.functional) (oneflow.Tensor method) sin() (in module oneflow) (oneflow.Tensor method) sin_() (in module oneflow) (oneflow.Tensor method) sinh() (in module oneflow) (oneflow.Tensor method) size() (oneflow.Tensor method) smooth_l1_loss() (in module oneflow.nn.functional) SmoothL1Loss (class in oneflow.nn) Softmax (class in oneflow.nn) softmax() (in module oneflow.nn.functional) (oneflow.Tensor method) Softplus (class in oneflow.nn) softplus() (in module oneflow.nn.functional) (oneflow.Tensor method) Softshrink (class in oneflow.nn) softshrink() (in module oneflow.nn.functional) Softsign (class in oneflow.nn) softsign() (in module oneflow.nn.functional) (oneflow.Tensor method) sort() (in module oneflow) (oneflow.Tensor method) sorted_indices (oneflow.nn.utils.rnn.PackedSequence attribute) sparse_softmax_cross_entropy() (in module oneflow.nn.functional) split() (in module oneflow) (oneflow.Tensor method) sqrt() (in module oneflow) (oneflow.Tensor method) square() (in module oneflow) (oneflow.Tensor method) squeeze() (in module oneflow) (oneflow.Tensor method) stack() (in module oneflow) state_dict() (oneflow.nn.Graph method) (oneflow.optim.Optimizer method) std() (in module oneflow) (oneflow.Tensor method) step() (oneflow.one_embedding.Ftrl method) (oneflow.optim.Optimizer method) StepLR (class in oneflow.optim.lr_scheduler) storage_offset() (oneflow.Tensor method) stride() (oneflow.Tensor method) sub() (in module oneflow) (oneflow.Tensor method) sub_() (oneflow.Tensor method) Subset (class in oneflow.utils.data) SubsetRandomSampler (class in oneflow.utils.data) sum() (in module oneflow) (oneflow.Tensor method) support_sparse() (oneflow.one_embedding.Ftrl property) swapaxes() (in module oneflow) (oneflow.Tensor method) swapdims() (in module oneflow) (oneflow.Tensor method) synchronize() (in module oneflow.cuda) T t() (in module oneflow) (oneflow.Tensor method) T() (oneflow.Tensor property) tan() (in module oneflow) (oneflow.Tensor method) Tanh (class in oneflow.nn) tanh() (in module oneflow) (in module oneflow.nn.functional) (oneflow.Tensor method) Tensor (class in oneflow) tensor() (in module oneflow) tensor_split() (in module oneflow) TensorDataset (class in oneflow.utils.data) tensordot() (in module oneflow) Threshold (class in oneflow.nn) threshold() (in module oneflow.nn.functional) tile() (in module oneflow) (oneflow.Tensor method) to() (oneflow.Tensor method) to_consistent() (oneflow.Tensor method) to_global() (oneflow.Tensor method) to_local() (oneflow.Tensor method) tolist() (oneflow.Tensor method) topk() (in module oneflow) (oneflow.Tensor method) transpose() (in module oneflow) (oneflow.Tensor method) tril() (in module oneflow) (oneflow.Tensor method) triplet_margin_loss() (in module oneflow.nn.functional) TripletMarginLoss (class in oneflow.nn) triu() (in module oneflow) (oneflow.Tensor method) trunc_normal_() (in module oneflow.nn.init) type() (oneflow.Tensor method) type_as() (oneflow.Tensor method) U unbind() (in module oneflow) (oneflow.Tensor method) Unfold (class in oneflow.nn) unfold() (oneflow.Tensor method) uniform_() (in module oneflow.nn.init) (oneflow.Tensor method) unsorted_indices (oneflow.nn.utils.rnn.PackedSequence attribute) unsqueeze() (in module oneflow) (oneflow.Tensor method) Upsample (class in oneflow.nn) UpsamplingBilinear2d (class in oneflow.nn) UpsamplingNearest2d (class in oneflow.nn) V var() (in module oneflow) (oneflow.Tensor method) vector_norm() (in module oneflow.linalg) view() (oneflow.Tensor method) view_as() (oneflow.Tensor method) vsplit() (in module oneflow) W weight (oneflow.nn.Conv1d attribute) (oneflow.nn.Conv3d attribute) (oneflow.nn.ConvTranspose1d attribute) (oneflow.nn.ConvTranspose2d attribute) (oneflow.nn.ConvTranspose3d attribute) weight_hh (oneflow.nn.GRUCell attribute) (oneflow.nn.LSTMCell attribute) (oneflow.nn.RNNCell attribute) weight_ih (oneflow.nn.GRUCell attribute) (oneflow.nn.LSTMCell attribute) (oneflow.nn.RNNCell attribute) weight_norm() (in module oneflow.nn.utils) where() (in module oneflow) (oneflow.Tensor method) X xavier_normal_() (in module oneflow.nn.init) xavier_uniform_() (in module oneflow.nn.init) Z zero_() (oneflow.Tensor method) zero_grad() (oneflow.optim.Optimizer method) ZeroPad2d (class in oneflow.nn) zeros() (in module oneflow) zeros_() (in module oneflow.nn.init) zeros_like() (in module oneflow)