Contents Menu Expand
OneFlow documentation
OneFlow documentation
  • Troubleshooting

OneFlow Python API

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

oneflow.Tensor.flip¶

Tensor.flip(dims)¶

See oneflow.flip()

Next
oneflow.Tensor.float
Previous
oneflow.Tensor.flatten
Copyright © 2020, OneFlow | Built with Sphinx and @pradyunsg's Furo theme. | Show Source