Index _ | A | B | C | D | E | F | G | H | I | K | L | M | N | O | P | Q | 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.BoolTensor method) (oneflow.ByteTensor method) (oneflow.CharTensor 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.DoubleTensor method) (oneflow.enable_grad method) (oneflow.FloatTensor method) (oneflow.HalfTensor method) (oneflow.inference_mode method) (oneflow.IntTensor method) (oneflow.LongTensor method) (oneflow.nn.COCOReader method) (oneflow.nn.CoinFlip method) (oneflow.nn.CropMirrorNormalize method) (oneflow.nn.FakeQuantization method) (oneflow.nn.Graph method) (oneflow.nn.MinMaxObserver method) (oneflow.nn.MovingAverageMinMaxObserver method) (oneflow.nn.OFRecordBytesDecoder method) (oneflow.nn.OFRecordImageDecoder method) (oneflow.nn.OFRecordImageDecoderRandomCrop method) (oneflow.nn.OFRecordRawDecoder method) (oneflow.nn.OFRecordReader method) (oneflow.nn.Parameter method) (oneflow.nn.QatConv1d method) (oneflow.nn.QatConv2d method) (oneflow.nn.QatConv3d method) (oneflow.nn.Quantization method) (oneflow.no_grad method) (oneflow.optim.Adagrad method) (oneflow.optim.Adam method) (oneflow.optim.AdamW method) (oneflow.optim.LAMB method) (oneflow.optim.LBFGS 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.graph_block.GraphModule property) Adagrad (class in oneflow.optim) Adam (class in oneflow.optim) AdamW (class in oneflow.optim) adaptive_avg_pool1d() (in module oneflow) (in module oneflow.nn.functional) adaptive_avg_pool2d() (in module oneflow) (in module oneflow.nn.functional) adaptive_avg_pool3d() (in module oneflow) (in module oneflow.nn.functional) adaptive_max_pool1d() (in module oneflow.nn.functional) adaptive_max_pool2d() (in module oneflow.nn.functional) adaptive_max_pool3d() (in module oneflow.nn.functional) AdaptiveAvgPool1d (class in oneflow.nn) AdaptiveAvgPool2d (class in oneflow.nn) AdaptiveAvgPool3d (class in oneflow.nn) AdaptiveMaxPool1d (class in oneflow.nn) AdaptiveMaxPool2d (class in oneflow.nn) AdaptiveMaxPool3d (class in oneflow.nn) add() (in module oneflow) (oneflow.Tensor method) add_() (oneflow.Tensor method) add_module() (oneflow.nn.Module method) add_optimizer() (oneflow.nn.Graph method) add_param_group() (oneflow.optim.Optimizer method) addcdiv() (in module oneflow) (oneflow.Tensor method) addcdiv_() (oneflow.Tensor 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) (in module oneflow.placement) (oneflow.Tensor method) all_device_placement() (in module oneflow.env) all_gather() (in module oneflow.comm) all_gather_into_tensor() (in module oneflow.comm) all_reduce() (in module oneflow.comm) all_to_all() (in module oneflow.comm) allclose() (in module oneflow) (oneflow.Tensor method) 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) apply() (oneflow.autograd.Function class method) (oneflow.nn.Module 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) (oneflow.Tensor method) as_strided_() (oneflow.Tensor method) 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) atleast_1d() (in module oneflow) atleast_2d() (in module oneflow) atleast_3d() (in module oneflow) 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.autograd.Function static method) (oneflow.Tensor method) baddbmm() (in module oneflow) (oneflow.Tensor method) barrier() (in module oneflow.comm) batch_align (in module oneflow.nn.image) batch_gather() (in module oneflow) batch_norm() (in module oneflow.nn.functional) 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) (oneflow.Tensor method) bernoulli_() (oneflow.Tensor method) bias (oneflow.nn.Conv1d attribute) (oneflow.nn.Conv3d attribute) (oneflow.nn.ConvTranspose1d attribute) (oneflow.nn.ConvTranspose2d attribute) (oneflow.nn.ConvTranspose3d attribute) (oneflow.nn.QatConv1d attribute) (oneflow.nn.QatConv3d 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) binary_cross_entropy() (in module oneflow.nn.functional) binary_cross_entropy_with_logits() (in module oneflow.nn.functional) bincount() (in module oneflow) (oneflow.Tensor method) bitwise_and() (in module oneflow) (oneflow.Tensor method) bitwise_not() (in module oneflow) bitwise_or() (in module oneflow) (oneflow.Tensor method) bitwise_xor() (in module oneflow) (oneflow.Tensor method) bmm() (in module oneflow) (oneflow.Tensor method) bool() (oneflow.Tensor method) BoolTensor (class in oneflow) (class in oneflow.cuda) broadcast() (in module oneflow.comm) broadcast_like() (in module oneflow) broadcast_shapes() (in module oneflow) broadcast_tensors() (in module oneflow) broadcast_to() (in module oneflow) (oneflow.Tensor method) buffers() (oneflow.nn.Module method) build() (oneflow.nn.Graph method) byte() (oneflow.Tensor method) ByteTensor (class in oneflow) (class in oneflow.cuda) C calculate_gain() (in module oneflow.nn.init) cast() (in module oneflow) (oneflow.Tensor method) cat() (in module oneflow) Categorical (class in oneflow.distributions) ceil() (in module oneflow) (oneflow.Tensor method) 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) (class in oneflow.cuda) children() (oneflow.nn.Module method) chunk() (in module oneflow) (oneflow.Tensor method) clamp() (in module oneflow) (oneflow.Tensor method) clamp_() (oneflow.Tensor method) clamp_max() (in module oneflow) clamp_min() (in module oneflow) 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) COCOReader (class in oneflow.nn) CoinFlip (class in oneflow.nn) column_stack() (in module oneflow) 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.nn.Module method) (oneflow.Tensor method) CropMirrorNormalize (class in oneflow.nn) cross() (in module oneflow) (in module oneflow.linalg) (oneflow.Tensor method) cross_entropy() (in module oneflow.nn.functional) CrossEntropyLoss (class in oneflow.nn) ctc_greedy_decoder() (in module oneflow.nn.functional) ctc_loss() (in module oneflow.nn.functional) CTCLoss (class in oneflow.nn) cuda() (oneflow.nn.Module method) (oneflow.Tensor method) cumprod() (in module oneflow) (oneflow.Tensor method) cumsum() (in module oneflow) (oneflow.Tensor method) current_device() (in module oneflow.cuda) current_global_mode (class in oneflow.utils.global_view) 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) deform_conv2d() (in module oneflow.nn.functional) det() (in module oneflow.linalg) 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) digamma() (in module oneflow.special) (oneflow.Tensor method) dim() (oneflow.Tensor method) DistributedDataParallel() (in module oneflow.nn.parallel) DistributedSampler (class in oneflow.utils.data.distributed) Distribution (class in oneflow.distributions) div() (in module oneflow) (oneflow.Tensor method) div_() (oneflow.Tensor method) dot() (in module oneflow) (oneflow.Tensor method) double() (oneflow.nn.Module method) (oneflow.Tensor method) DoubleTensor (class in oneflow) (class in oneflow.cuda) download_url_to_file() (in module oneflow.hub) Dropout (class in oneflow.nn) dropout() (in module oneflow.nn.functional) Dropout1d (class in oneflow.nn) dropout1d() (in module oneflow.nn.functional) Dropout2d (class in oneflow.nn) dropout2d() (in module oneflow.nn.functional) Dropout3d (class in oneflow.nn) dropout3d() (in module oneflow.nn.functional) dstack() (in module oneflow) 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) empty_cache() (in module oneflow.cuda) empty_like() (in module oneflow) enable_amp() (oneflow.nn.graph.graph_config.GraphConfig method) enable_auto_memory() (oneflow.nn.graph.graph_config.GraphConfig method) enable_auto_parallel() (oneflow.nn.graph.graph_config.GraphConfig method) enable_auto_parallel_ignore_user_sbp_config() (oneflow.nn.graph.graph_config.GraphConfig method) enable_auto_parallel_sbp_collector() (oneflow.nn.graph.graph_config.GraphConfig method) enable_auto_parallel_trunk_algo() (oneflow.nn.graph.graph_config.GraphConfig method) enable_compress_memory() (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_straighten_algorithm() (oneflow.nn.graph.graph_config.GraphConfig method) enable_zero() (oneflow.nn.graph.graph_config.GraphConfig method) eq() (in module oneflow) (oneflow.Tensor method) equal() (in module oneflow) (oneflow.Tensor method) erf() (in module oneflow) (in module oneflow.special) (oneflow.Tensor method) erfc() (in module oneflow) (in module oneflow.special) (oneflow.Tensor method) erfinv() (in module oneflow) (in module oneflow.special) (oneflow.Tensor method) erfinv_() (oneflow.Tensor method) eval() (oneflow.nn.Module method) exp() (in module oneflow) (oneflow.Tensor method) exp2() (in module oneflow.special) (oneflow.Tensor method) expand() (in module oneflow) (oneflow.Tensor method) expand_as() (oneflow.Tensor method) expm1() (in module oneflow) (in module oneflow.special) (oneflow.Tensor method) ExponentialLR (class in oneflow.optim.lr_scheduler) extra_repr() (oneflow.nn.Module method) eye() (in module oneflow) F FakeQuantization (class in oneflow.nn) 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.nn.Module method) (oneflow.Tensor method) FloatTensor (class in oneflow) (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) fold() (in module oneflow.nn.functional) forward() (oneflow.autograd.Function static method) (oneflow.nn.Module method) (oneflow.one_embedding.MultiTableEmbedding method) (oneflow.one_embedding.MultiTableMultiColumnEmbedding method) frac() (in module oneflow) (oneflow.Tensor method) frac_() (in module oneflow) (oneflow.Tensor method) from_numpy() (in module oneflow) from_torch() (in module oneflow.utils.tensor) 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) gather_nd() (in module oneflow) ge() (in module oneflow) (oneflow.Tensor method) GELU (class in oneflow.nn) gelu() (in module oneflow) (in module oneflow.nn.functional) get_default_dtype() (in module oneflow) get_device() (oneflow.Tensor method) get_device_capability() (in module oneflow.cuda) get_device_name() (in module oneflow.cuda) get_device_properties() (in module oneflow.cuda) get_dir() (in module oneflow.hub) 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) (in module oneflow.cuda) get_rng_state_all() (in module oneflow.cuda) get_world_size() (in module oneflow.env) global_mode (class in oneflow.utils.global_view) 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) gt_() (oneflow.Tensor method) gumbel_softmax() (in module oneflow.nn.functional) H half() (oneflow.Tensor method) HalfTensor (class in oneflow) (class in oneflow.cuda) hann_window() (in module oneflow) 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) help() (in module oneflow.hub) hsplit() (in module oneflow) hstack() (in module oneflow) I Identity (class in oneflow.nn) in_top_k() (in module oneflow) (oneflow.Tensor method) index_add() (in module oneflow) (oneflow.Tensor method) index_add_() (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) (class in oneflow.cuda) inv() (in module oneflow.linalg) inverse() (oneflow.Tensor method) 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_offloaded() (oneflow.Tensor method) is_pinned() (oneflow.Tensor method) is_tensor() (in module oneflow) isclose() (in module oneflow) (oneflow.Tensor method) 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 l1_loss() (in module oneflow.nn.functional) 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) LBFGS (class in oneflow.optim) le() (in module oneflow) (oneflow.Tensor method) leaky_relu() (in module oneflow.nn.functional) LeakyReLU (class in oneflow.nn) lerp() (in module oneflow) (oneflow.Tensor method) lerp_() (in module oneflow) (oneflow.Tensor method) Linear (class in oneflow.nn) linear() (in module oneflow.nn.functional) LinearLR (class in oneflow.optim.lr_scheduler) linspace() (in module oneflow) list() (in module oneflow.hub) load() (in module oneflow) (in module oneflow.hub) (oneflow.Tensor method) load_snapshot() (oneflow.one_embedding.MultiTableEmbedding method) (oneflow.one_embedding.MultiTableMultiColumnEmbedding method) load_state_dict() (oneflow.nn.Graph method) (oneflow.nn.Module method) (oneflow.optim.Optimizer method) load_state_dict_from_url() (in module oneflow.hub) local_to_global() (oneflow.Tensor method) log() (in module oneflow) (oneflow.Tensor method) log10() (in module oneflow) (oneflow.Tensor method) log1p() (in module oneflow) (in module oneflow.special) (oneflow.Tensor method) log2() (in module oneflow) (oneflow.Tensor method) log_softmax() (in module oneflow.nn.functional) (in module oneflow.special) 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) logspace() (in module oneflow) logsumexp() (in module oneflow) (in module oneflow.special) (oneflow.Tensor method) long() (oneflow.Tensor method) LongTensor (class in oneflow) (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._oneflow_internal.autograd.Function.FunctionCtx method) masked_fill() (in module oneflow) (oneflow.Tensor 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) max_unpool1d() (in module oneflow.nn.functional) max_unpool2d() (in module oneflow.nn.functional) max_unpool3d() (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) MaxUnpool1d (class in oneflow.nn) MaxUnpool2d (class in oneflow.nn) MaxUnpool3d (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) MinMaxObserver (class in oneflow.nn) Mish (class in oneflow.nn) mish() (in module oneflow) (in module oneflow.nn.functional) (oneflow.Tensor method) mm() (in module oneflow) (oneflow.Tensor method) mode() (in module oneflow) (oneflow.Tensor method) module oneflow.hub oneflow.one_embedding oneflow.utils.data Module (class in oneflow.nn) ModuleDict (class in oneflow.nn) ModuleList (class in oneflow.nn) modules() (oneflow.nn.Module method) movedim() (in module oneflow) MovingAverageMinMaxObserver (class in oneflow.nn) mse_loss() (in module oneflow.nn.functional) MSELoss (class in oneflow.nn) mul() (in module oneflow) (oneflow.Tensor method) mul_() (oneflow.Tensor method) multinomial() (in module oneflow) 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) named_buffers() (oneflow.nn.Module method) named_children() (oneflow.nn.Module method) named_modules() (oneflow.nn.Module method) named_parameters() (oneflow.nn.Module method) nansum() (in module oneflow) (oneflow.Tensor method) 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_full() (oneflow.Tensor method) new_ones() (in module oneflow) (oneflow.Tensor method) new_tensor() (oneflow.Tensor method) new_zeros() (oneflow.Tensor method) NLLLoss (class in oneflow.nn) nms() (in module oneflow) (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 offload() (oneflow.Tensor method) OFRecordBytesDecoder (class in oneflow.nn) OFRecordImageDecoder (class in oneflow.nn) OFRecordImageDecoderRandomCrop (class in oneflow.nn) OFRecordRawDecoder (class in oneflow.nn) OFRecordReader (class in oneflow.nn) one_hot() (in module oneflow.nn.functional) oneflow.finfo (class in oneflow) oneflow.hub module oneflow.iinfo (class in oneflow) 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) pairwise_distance() (in module oneflow.nn.functional) PairwiseDistance (class in oneflow.nn) Parameter (class in oneflow.nn) ParameterDict (class in oneflow.nn) ParameterList (class in oneflow.nn) parameters() (oneflow.nn.Module method) 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) Q QatConv1d (class in oneflow.nn) QatConv2d (class in oneflow.nn) QatConv3d (class in oneflow.nn) quantile() (in module oneflow) (oneflow.Tensor method) Quantization (class in oneflow.nn) quick_gelu() (in module oneflow) (in module oneflow.nn.functional) QuickGELU (class in oneflow.nn) R rand() (in module oneflow) randint() (in module oneflow) randint_like() (in module oneflow) randn() (in module oneflow) randn_like() (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) reduce_scatter_tensor() (in module oneflow.comm) ReduceLROnPlateau (class in oneflow.optim.lr_scheduler) ReflectionPad1d (class in oneflow.nn) ReflectionPad2d (class in oneflow.nn) register_backward_hook() (oneflow.nn.Module method) register_buffer() (oneflow.nn.Module method) register_forward_hook() (oneflow.nn.Module method) register_forward_pre_hook() (oneflow.nn.Module method) register_full_backward_hook() (oneflow.nn.Module method) register_hook() (oneflow.Tensor method) register_parameter() (oneflow.nn.Module method) register_state_dict_pre_hook() (oneflow.nn.Module 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() (in module oneflow) (oneflow.Tensor method) repeat_interleave() (in module oneflow) (oneflow.Tensor method) ReplicationPad1d (class in oneflow.nn) ReplicationPad2d (class in oneflow.nn) requires_grad (oneflow.Tensor attribute) requires_grad_() (oneflow.nn.Module method) (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) RMSLayerNorm (class in oneflow.nn) RMSNorm (class in oneflow.nn) RMSprop (class in oneflow.optim) RNN (class in oneflow.nn) RNNCell (class in oneflow.nn) roc_auc_score() (in module oneflow) roll() (in module oneflow) (oneflow.Tensor method) round() (in module oneflow) (in module oneflow.special) (oneflow.Tensor method) round_() (in module oneflow) (oneflow.Tensor method) row_stack() (in module oneflow) rsqrt() (in module oneflow) (oneflow.Tensor method) S Sampler (class in oneflow.utils.data) save() (in module oneflow) save_for_backward() (oneflow._oneflow_internal.autograd.Function.FunctionCtx method) save_snapshot() (oneflow.one_embedding.MultiTableEmbedding method) (oneflow.one_embedding.MultiTableMultiColumnEmbedding method) saved_tensors (oneflow._oneflow_internal.autograd.Function.FunctionCtx attribute) sbp (class in oneflow.sbp) (oneflow.Tensor attribute) scatter() (in module oneflow) (in module oneflow.comm) (oneflow.Tensor method) scatter_() (oneflow.Tensor method) scatter_add() (in module oneflow) (oneflow.Tensor method) scatter_add_() (oneflow.Tensor method) 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) (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_auto_parallel_computation_cost_ratio() (oneflow.nn.graph.graph_config.GraphConfig method) set_auto_parallel_wait_time() (oneflow.nn.graph.graph_config.GraphConfig method) set_default_dtype() (in module oneflow) set_default_tensor_type() (in module oneflow) set_device() (in module oneflow.cuda) set_dir() (in module oneflow.hub) 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) (in module oneflow.cuda) set_rng_state_all() (in module oneflow.cuda) set_stage() (oneflow.nn.graph.graph_block.GraphModule 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) (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) slice() (in module oneflow) slice_update() (in module oneflow) smooth_l1_loss() (in module oneflow.nn.functional) SmoothL1Loss (class in oneflow.nn) Softmax (class in oneflow.nn) softmax() (in module oneflow) (in module oneflow.nn.functional) (in module oneflow.special) (oneflow.Tensor method) Softplus (class in oneflow.nn) softplus() (in module oneflow) (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) (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) square_relu() (in module oneflow) (in module oneflow.nn.functional) SquareReLU (class in oneflow.nn) squeeze() (in module oneflow) (oneflow.Tensor method) squeeze_() (oneflow.Tensor method) stack() (in module oneflow) state_dict() (oneflow.nn.Graph method) (oneflow.nn.Module 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) SyncBatchNorm (class in oneflow.nn) 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_scatter_nd_update() (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.nn.Module method) (oneflow.Tensor method) to_consistent() (oneflow.Tensor method) to_global() (in module oneflow.utils.global_view) (oneflow.Tensor method) to_local() (in module oneflow.utils.global_view) (oneflow.Tensor method) to_torch() (in module oneflow.utils.tensor) tolist() (oneflow.Tensor method) topk() (in module oneflow) (oneflow.Tensor method) train() (oneflow.nn.Module 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() (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() (in module oneflow.nn.functional) (oneflow.Tensor method) uniform_() (in module oneflow.nn.init) (oneflow.Tensor method) unique() (in module oneflow) (oneflow.Tensor method) unsorted_indices (oneflow.nn.utils.rnn.PackedSequence attribute) unsqueeze() (in module oneflow) (oneflow.Tensor method) unsqueeze_() (oneflow.Tensor method) Upsample (class in oneflow.nn) upsample (in module oneflow.nn.functional) 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) vstack() (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) (oneflow.nn.QatConv1d attribute) (oneflow.nn.QatConv3d 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.nn.Module method) (oneflow.optim.Optimizer method) ZeroPad2d (class in oneflow.nn) zeros() (in module oneflow) zeros_() (in module oneflow.nn.init) zeros_like() (in module oneflow) zeta() (in module oneflow.special)