Dragon API Dragon v0.3.0 Python Index _ | A | B | C | D | E | F | G | H | I | K | L | M | N | O | P | R | S | T | U | V | W | X | Z _ __add__() (dragon.Tensor method) __and__() (dragon.Tensor method) __call__() (dragon.vm.tensorflow.keras.initializers.Constant method) (dragon.vm.tensorflow.keras.initializers.Initializer method) (dragon.vm.tensorflow.keras.initializers.Ones method) (dragon.vm.tensorflow.keras.initializers.RandomNormal method) (dragon.vm.tensorflow.keras.initializers.RandomUniform method) (dragon.vm.tensorflow.keras.initializers.TruncatedNormal method) (dragon.vm.tensorflow.keras.initializers.VarianceScaling method) (dragon.vm.tensorflow.keras.initializers.Zeros method) (dragon.vm.tensorflow.keras.losses.Loss method) (dragon.vm.tensorflow.keras.regularizers.Regularizer method) __eq__() (dragon.Tensor method) __float__() (dragon.Tensor method) __ge__() (dragon.Tensor method) __getitem__() (dragon.Tensor method) __gt__() (dragon.Tensor method) __iadd__() (dragon.Tensor method) __iand__() (dragon.Tensor method) __imul__() (dragon.Tensor method) __init__() (dragon.cuda.Stream method) (dragon.DeviceSpec method) (dragon.GradientTape method) (dragon.nn.GRU method) (dragon.nn.LSTM method) (dragon.nn.RNN method) (dragon.onnx.BackendRep method) (dragon.optimizers.Adam method) (dragon.optimizers.AdamW method) (dragon.optimizers.Optimizer method) (dragon.optimizers.RMSprop method) (dragon.optimizers.SGD method) (dragon.Tensor method) (dragon.vm.dali.Iterator method) (dragon.vm.dali.Pipeline method) (dragon.vm.tensorflow.dtypes.DType method) (dragon.vm.tensorflow.GradientTape method) (dragon.vm.tensorflow.keras.initializers.Constant method) (dragon.vm.tensorflow.keras.initializers.GlorotNormal method) (dragon.vm.tensorflow.keras.initializers.GlorotUniform method) (dragon.vm.tensorflow.keras.initializers.Ones method) (dragon.vm.tensorflow.keras.initializers.RandomNormal method) (dragon.vm.tensorflow.keras.initializers.RandomUniform method) (dragon.vm.tensorflow.keras.initializers.TruncatedNormal method) (dragon.vm.tensorflow.keras.initializers.VarianceScaling method) (dragon.vm.tensorflow.keras.initializers.Zeros method) (dragon.vm.tensorflow.keras.layers.Activation method) (dragon.vm.tensorflow.keras.layers.Add method) (dragon.vm.tensorflow.keras.layers.AveragePooling1D method) (dragon.vm.tensorflow.keras.layers.AveragePooling2D method) (dragon.vm.tensorflow.keras.layers.AveragePooling3D method) (dragon.vm.tensorflow.keras.layers.BatchNormalization method) (dragon.vm.tensorflow.keras.layers.Concatenate method) (dragon.vm.tensorflow.keras.layers.Conv1D method) (dragon.vm.tensorflow.keras.layers.Conv1DTranspose method) (dragon.vm.tensorflow.keras.layers.Conv2D method) (dragon.vm.tensorflow.keras.layers.Conv2DTranspose method) (dragon.vm.tensorflow.keras.layers.Conv3D method) (dragon.vm.tensorflow.keras.layers.Conv3DTranspose method) (dragon.vm.tensorflow.keras.layers.Dense method) (dragon.vm.tensorflow.keras.layers.DepthwiseConv2D method) (dragon.vm.tensorflow.keras.layers.Dropout method) (dragon.vm.tensorflow.keras.layers.ELU method) (dragon.vm.tensorflow.keras.layers.Flatten method) (dragon.vm.tensorflow.keras.layers.GlobalAveragePooling1D method) (dragon.vm.tensorflow.keras.layers.GlobalAveragePooling2D method) (dragon.vm.tensorflow.keras.layers.GlobalAveragePooling3D method) (dragon.vm.tensorflow.keras.layers.GlobalMaxPooling1D method) (dragon.vm.tensorflow.keras.layers.GlobalMaxPooling2D method) (dragon.vm.tensorflow.keras.layers.GlobalMaxPooling3D method) (dragon.vm.tensorflow.keras.layers.Layer method) (dragon.vm.tensorflow.keras.layers.LayerNormalization method) (dragon.vm.tensorflow.keras.layers.LeakyReLU method) (dragon.vm.tensorflow.keras.layers.Maximum method) (dragon.vm.tensorflow.keras.layers.MaxPooling1D method) (dragon.vm.tensorflow.keras.layers.MaxPooling2D method) (dragon.vm.tensorflow.keras.layers.MaxPooling3D method) (dragon.vm.tensorflow.keras.layers.Minimum method) (dragon.vm.tensorflow.keras.layers.Multiply method) (dragon.vm.tensorflow.keras.layers.Permute method) (dragon.vm.tensorflow.keras.layers.ReLU method) (dragon.vm.tensorflow.keras.layers.Reshape method) (dragon.vm.tensorflow.keras.layers.SELU method) (dragon.vm.tensorflow.keras.layers.Softmax method) (dragon.vm.tensorflow.keras.layers.Subtract method) (dragon.vm.tensorflow.keras.layers.UpSampling1D method) (dragon.vm.tensorflow.keras.layers.UpSampling2D method) (dragon.vm.tensorflow.keras.layers.UpSampling3D method) (dragon.vm.tensorflow.keras.layers.ZeroPadding1D method) (dragon.vm.tensorflow.keras.layers.ZeroPadding2D method) (dragon.vm.tensorflow.keras.layers.ZeroPadding3D method) (dragon.vm.tensorflow.keras.losses.BinaryCrossentropy method) (dragon.vm.tensorflow.keras.losses.CategoricalCrossentropy method) (dragon.vm.tensorflow.keras.losses.Loss method) (dragon.vm.tensorflow.keras.losses.MeanAbsoluteError method) (dragon.vm.tensorflow.keras.losses.MeanSquaredError method) (dragon.vm.tensorflow.keras.losses.SparseCategoricalCrossentropy method) (dragon.vm.tensorflow.keras.optimizers.Adam method) (dragon.vm.tensorflow.keras.optimizers.Optimizer method) (dragon.vm.tensorflow.keras.optimizers.RMSprop method) (dragon.vm.tensorflow.keras.optimizers.SGD method) (dragon.vm.tensorflow.keras.regularizers.L1 method) (dragon.vm.tensorflow.keras.regularizers.L1L2 method) (dragon.vm.tensorflow.keras.regularizers.L2 method) (dragon.vm.tensorflow.keras.Sequential method) (dragon.vm.tensorflow.Module method) (dragon.vm.tensorflow.TensorShape method) (dragon.vm.tensorflow.TensorSpec method) (dragon.vm.tensorrt.Binding method) (dragon.vm.tensorrt.Engine method) (dragon.vm.tensorrt.onnx.BackendRep method) (dragon.vm.torch.device method) (dragon.vm.torch.dtype method) (dragon.vm.torch.enable_grad method) (dragon.vm.torch.nn.AdaptiveAvgPool1d method) (dragon.vm.torch.nn.AdaptiveAvgPool2d method) (dragon.vm.torch.nn.AdaptiveAvgPool3d method) (dragon.vm.torch.nn.AdaptiveMaxPool1d method) (dragon.vm.torch.nn.AdaptiveMaxPool2d method) (dragon.vm.torch.nn.AdaptiveMaxPool3d method) (dragon.vm.torch.nn.Affine method) (dragon.vm.torch.nn.AvgPool1d method) (dragon.vm.torch.nn.AvgPool2d method) (dragon.vm.torch.nn.AvgPool3d method) (dragon.vm.torch.nn.BatchNorm1d method) (dragon.vm.torch.nn.BatchNorm2d method) (dragon.vm.torch.nn.BatchNorm3d method) (dragon.vm.torch.nn.BCEWithLogitsLoss method) (dragon.vm.torch.nn.ChannelShuffle method) (dragon.vm.torch.nn.ConstantPad1d method) (dragon.vm.torch.nn.ConstantPad2d method) (dragon.vm.torch.nn.ConstantPad3d method) (dragon.vm.torch.nn.Conv1d method) (dragon.vm.torch.nn.Conv2d method) (dragon.vm.torch.nn.Conv3d method) (dragon.vm.torch.nn.ConvTranspose1d method) (dragon.vm.torch.nn.ConvTranspose2d method) (dragon.vm.torch.nn.ConvTranspose3d method) (dragon.vm.torch.nn.CosineSimilarity method) (dragon.vm.torch.nn.CrossEntropyLoss method) (dragon.vm.torch.nn.CTCLoss method) (dragon.vm.torch.nn.DepthwiseConv2d method) (dragon.vm.torch.nn.DropBlock2d method) (dragon.vm.torch.nn.Dropout method) (dragon.vm.torch.nn.DropPath method) (dragon.vm.torch.nn.ELU method) (dragon.vm.torch.nn.Flatten method) (dragon.vm.torch.nn.GELU method) (dragon.vm.torch.nn.GroupNorm method) (dragon.vm.torch.nn.GRU method) (dragon.vm.torch.nn.GumbelSoftmax method) (dragon.vm.torch.nn.Hardsigmoid method) (dragon.vm.torch.nn.Hardswish method) (dragon.vm.torch.nn.Identity method) (dragon.vm.torch.nn.KLDivLoss method) (dragon.vm.torch.nn.L1Loss method) (dragon.vm.torch.nn.LayerNorm method) (dragon.vm.torch.nn.LeakyReLU method) (dragon.vm.torch.nn.Linear method) (dragon.vm.torch.nn.LocalResponseNorm method) (dragon.vm.torch.nn.LogSoftmax method) (dragon.vm.torch.nn.LSTM method) (dragon.vm.torch.nn.LSTMCell method) (dragon.vm.torch.nn.MaxPool1d method) (dragon.vm.torch.nn.MaxPool2d method) (dragon.vm.torch.nn.MaxPool3d method) (dragon.vm.torch.nn.Module method) (dragon.vm.torch.nn.ModuleList method) (dragon.vm.torch.nn.MSELoss method) (dragon.vm.torch.nn.MultiheadAttention method) (dragon.vm.torch.nn.NLLLoss method) (dragon.vm.torch.nn.Parameter method) (dragon.vm.torch.nn.PixelShuffle method) (dragon.vm.torch.nn.PixelUnshuffle method) (dragon.vm.torch.nn.PReLU method) (dragon.vm.torch.nn.ReflectionPad1d method) (dragon.vm.torch.nn.ReflectionPad2d method) (dragon.vm.torch.nn.ReflectionPad3d method) (dragon.vm.torch.nn.ReLU method) (dragon.vm.torch.nn.ReLU6 method) (dragon.vm.torch.nn.ReplicationPad1d method) (dragon.vm.torch.nn.ReplicationPad2d method) (dragon.vm.torch.nn.ReplicationPad3d method) (dragon.vm.torch.nn.RNN method) (dragon.vm.torch.nn.SELU method) (dragon.vm.torch.nn.Sequential method) (dragon.vm.torch.nn.Sigmoid method) (dragon.vm.torch.nn.SigmoidFocalLoss method) (dragon.vm.torch.nn.SiLU method) (dragon.vm.torch.nn.SmoothL1Loss method) (dragon.vm.torch.nn.Softmax method) (dragon.vm.torch.nn.SyncBatchNorm method) (dragon.vm.torch.nn.Tanh method) (dragon.vm.torch.nn.TransformerDecoder method) (dragon.vm.torch.nn.TransformerDecoderLayer method) (dragon.vm.torch.nn.TransformerEncoder method) (dragon.vm.torch.nn.TransformerEncoderLayer method) (dragon.vm.torch.nn.Unfold method) (dragon.vm.torch.nn.Upsample method) (dragon.vm.torch.nn.UpsamplingBilinear2d method) (dragon.vm.torch.nn.UpsamplingNearest2d method) (dragon.vm.torch.nn.ZeroPad2d method) (dragon.vm.torch.no_grad method) (dragon.vm.torch.optim.Adam method) (dragon.vm.torch.optim.AdamW method) (dragon.vm.torch.optim.LARS method) (dragon.vm.torch.optim.Optimizer method) (dragon.vm.torch.optim.RMSprop method) (dragon.vm.torch.optim.SGD method) (dragon.vm.torch.set_grad_enabled method) (dragon.vm.torch.Size method) (dragon.vm.torch.utils.checkpoint.no_checkpoint method) (dragon.Workspace method) __int__() (dragon.Tensor method) __invert__() (dragon.Tensor method) __ior__() (dragon.Tensor method) __isub__() (dragon.Tensor method) __itruediv__() (dragon.Tensor method) __ixor__() (dragon.Tensor method) __le__() (dragon.Tensor method) __lt__() (dragon.Tensor method) __matmul__() (dragon.Tensor method) __mul__() (dragon.Tensor method) __ne__() (dragon.Tensor method) __neg__() (dragon.Tensor method) __new__() (dragon.vm.dali.ops.BbFlip static method) (dragon.vm.dali.ops.BBoxPaste static method) (dragon.vm.dali.ops.Brightness static method) (dragon.vm.dali.ops.BrightnessContrast static method) (dragon.vm.dali.ops.Cast static method) (dragon.vm.dali.ops.CoinFlip static method) (dragon.vm.dali.ops.ColorSpaceConversion static method) (dragon.vm.dali.ops.ColorTwist static method) (dragon.vm.dali.ops.Contrast static method) (dragon.vm.dali.ops.CropMirrorNormalize static method) (dragon.vm.dali.ops.Erase static method) (dragon.vm.dali.ops.ExternalSource static method) (dragon.vm.dali.ops.GaussianBlur static method) (dragon.vm.dali.ops.Hsv static method) (dragon.vm.dali.ops.ImageDecoder static method) (dragon.vm.dali.ops.ImageDecoderRandomCrop static method) (dragon.vm.dali.ops.Normalize static method) (dragon.vm.dali.ops.Pad static method) (dragon.vm.dali.ops.Paste static method) (dragon.vm.dali.ops.RandomBBoxCrop static method) (dragon.vm.dali.ops.RandomResizedCrop static method) (dragon.vm.dali.ops.Reshape static method) (dragon.vm.dali.ops.Resize static method) (dragon.vm.dali.ops.Rotate static method) (dragon.vm.dali.ops.Slice static method) (dragon.vm.dali.ops.TFRecordReader static method) (dragon.vm.dali.ops.Uniform static method) (dragon.vm.dali.ops.WarpAffine static method) __next__() (dragon.vm.dali.Iterator method) __or__() (dragon.Tensor method) __radd__() (dragon.Tensor method) __rand__() (dragon.Tensor method) __rmul__() (dragon.Tensor method) __ror__() (dragon.Tensor method) __rsub__() (dragon.Tensor method) __rtruediv__() (dragon.Tensor method) __rxor__() (dragon.Tensor method) __setitem__() (dragon.Tensor method) __sub__() (dragon.Tensor method) __truediv__() (dragon.Tensor method) __xor__() (dragon.Tensor method) A abs() (dragon.vm.torch.Tensor method) (in module dragon.math) (in module dragon.vm.tensorflow.math) (in module dragon.vm.torch) accuracy() (in module dragon.metrics) Activation (class in dragon.vm.tensorflow.keras.layers) Adam (class in dragon.optimizers) (class in dragon.vm.tensorflow.keras.optimizers) (class in dragon.vm.torch.optim) AdamW (class in dragon.optimizers) (class in dragon.vm.torch.optim) adaptive_avg_pool1d() (in module dragon.vm.torch.nn.functional) adaptive_avg_pool2d() (in module dragon.vm.torch.nn.functional) adaptive_avg_pool3d() (in module dragon.vm.torch.nn.functional) adaptive_max_pool1d() (in module dragon.vm.torch.nn.functional) adaptive_max_pool2d() (in module dragon.vm.torch.nn.functional) adaptive_max_pool3d() (in module dragon.vm.torch.nn.functional) AdaptiveAvgPool1d (class in dragon.vm.torch.nn) AdaptiveAvgPool2d (class in dragon.vm.torch.nn) AdaptiveAvgPool3d (class in dragon.vm.torch.nn) AdaptiveMaxPool1d (class in dragon.vm.torch.nn) AdaptiveMaxPool2d (class in dragon.vm.torch.nn) AdaptiveMaxPool3d (class in dragon.vm.torch.nn) Add (class in dragon.vm.tensorflow.keras.layers) add() (dragon.vm.tensorflow.keras.Sequential method) (dragon.vm.torch.Tensor method) (in module dragon.math) (in module dragon.vm.tensorflow.math) (in module dragon.vm.torch) add_() (dragon.vm.torch.Tensor method) add_module() (dragon.vm.torch.nn.Module method) add_n() (in module dragon.vm.tensorflow.math) add_param_group() (dragon.vm.torch.optim.Optimizer method) add_weight() (dragon.vm.tensorflow.keras.layers.Layer method) addmm() (dragon.vm.torch.Tensor method) (in module dragon.vm.torch) Affine (class in dragon.vm.torch.nn) affine() (in module dragon.math) (in module dragon.vm.torch.nn.functional) all_reduce() (in module dragon.distributed) (in module dragon.vm.torch.distributed) apply() (dragon.vm.torch.nn.Module method) (dragon.vm.torch.nn.ModuleList method) apply_gradients() (dragon.optimizers.Optimizer method) (dragon.vm.tensorflow.keras.optimizers.Optimizer method) arange() (in module dragon.vm.torch) argmax() (dragon.vm.torch.Tensor method) (in module dragon.math) (in module dragon.vm.tensorflow.math) (in module dragon.vm.torch) argmin() (dragon.vm.torch.Tensor method) (in module dragon.math) (in module dragon.vm.tensorflow.math) (in module dragon.vm.torch) argsort() (dragon.vm.torch.Tensor method) (in module dragon) (in module dragon.vm.tensorflow) (in module dragon.vm.torch) as_datatype_enum (dragon.vm.tensorflow.dtypes.DType attribute) as_default() (dragon.Workspace method) as_dtype() (in module dragon.vm.tensorflow.dtypes) as_list() (dragon.vm.tensorflow.TensorShape method) as_numpy_dtype (dragon.vm.tensorflow.dtypes.DType attribute) assign() (in module dragon) astype() (dragon.Tensor method) atan2() (dragon.vm.torch.Tensor method) (in module dragon.math) (in module dragon.vm.tensorflow.math) (in module dragon.vm.torch) AveragePooling1D (class in dragon.vm.tensorflow.keras.layers) AveragePooling2D (class in dragon.vm.tensorflow.keras.layers) AveragePooling3D (class in dragon.vm.tensorflow.keras.layers) avg_pool() (in module dragon.vm.tensorflow.nn) avg_pool1d() (in module dragon.vm.tensorflow.nn) (in module dragon.vm.torch.nn.functional) avg_pool2d() (in module dragon.vm.tensorflow.nn) (in module dragon.vm.torch.nn.functional) avg_pool3d() (in module dragon.vm.tensorflow.nn) (in module dragon.vm.torch.nn.functional) AvgPool1d (class in dragon.vm.torch.nn) AvgPool2d (class in dragon.vm.torch.nn) AvgPool3d (class in dragon.vm.torch.nn) B BackendRep (class in dragon.onnx) (class in dragon.vm.tensorrt.onnx) backward() (dragon.vm.torch.Tensor method) (in module dragon.vm.torch.autograd) baddbmm() (dragon.vm.torch.Tensor method) (in module dragon.vm.torch) baddbmm_() (dragon.vm.torch.Tensor method) base_dtype (dragon.vm.tensorflow.dtypes.DType attribute) batch_norm() (in module dragon.nn) (in module dragon.vm.torch.nn.functional) batch_size (dragon.vm.dali.Iterator attribute) BatchNorm1d (class in dragon.vm.torch.nn) BatchNorm2d (class in dragon.vm.torch.nn) BatchNorm3d (class in dragon.vm.torch.nn) BatchNormalization (class in dragon.vm.tensorflow.keras.layers) BbFlip (class in dragon.vm.dali.ops) BBoxPaste (class in dragon.vm.dali.ops) BCEWithLogitsLoss (class in dragon.vm.torch.nn) bias_add() (in module dragon.nn) (in module dragon.vm.tensorflow.nn) binary_cross_entropy_with_logits() (in module dragon.vm.torch.nn.functional) binary_crossentropy() (in module dragon.vm.tensorflow.keras.losses) BinaryCrossentropy (class in dragon.vm.tensorflow.keras.losses) Binding (class in dragon.vm.tensorrt) bitwise_and() (dragon.vm.torch.Tensor method) (in module dragon.bitwise) (in module dragon.vm.tensorflow.bitwise) (in module dragon.vm.torch) bitwise_and_() (dragon.vm.torch.Tensor method) bitwise_not() (dragon.vm.torch.Tensor method) (in module dragon.vm.torch) bitwise_not_() (dragon.vm.torch.Tensor method) bitwise_or() (dragon.vm.torch.Tensor method) (in module dragon.bitwise) (in module dragon.vm.tensorflow.bitwise) (in module dragon.vm.torch) bitwise_or_() (dragon.vm.torch.Tensor method) bitwise_xor() (dragon.vm.torch.Tensor method) (in module dragon.bitwise) (in module dragon.vm.tensorflow.bitwise) (in module dragon.vm.torch) bitwise_xor_() (dragon.vm.torch.Tensor method) bmm() (dragon.vm.torch.Tensor method) (in module dragon.vm.torch) bool() (dragon.vm.torch.Tensor method) bool_() (dragon.vm.torch.Tensor method) boolean_mask() (in module dragon) Brightness (class in dragon.vm.dali.ops) BrightnessContrast (class in dragon.vm.dali.ops) broadcast() (in module dragon.distributed) (in module dragon.vm.torch.distributed) broadcast_to() (in module dragon) (in module dragon.vm.tensorflow) buffers() (dragon.vm.torch.nn.Module method) build() (dragon.vm.dali.Pipeline method) (dragon.vm.tensorflow.keras.layers.Layer method) byte() (dragon.vm.torch.Tensor method) byte_() (dragon.vm.torch.Tensor method) C calculate_gain() (in module dragon.vm.torch.nn.init) call() (dragon.vm.tensorflow.keras.layers.Layer method) Cast (class in dragon.vm.dali.ops) cast() (in module dragon) (in module dragon.vm.tensorflow) cat() (in module dragon.vm.torch) categorical_crossentropy() (in module dragon.vm.tensorflow.keras.losses) CategoricalCrossentropy (class in dragon.vm.tensorflow.keras.losses) ceil() (dragon.vm.torch.Tensor method) (in module dragon.math) (in module dragon.vm.tensorflow.math) (in module dragon.vm.torch) ceil_() (dragon.vm.torch.Tensor method) channel_norm() (in module dragon.nn) (in module dragon.vm.torch.nn.functional) channel_shuffle() (in module dragon.nn) (in module dragon.vm.torch.nn.functional) ChannelShuffle (class in dragon.vm.torch.nn) char() (dragon.vm.torch.Tensor method) char_() (dragon.vm.torch.Tensor method) checkpoint() (in module dragon.vm.torch.utils.checkpoint) checkpoint_sequential() (in module dragon.vm.torch.utils.checkpoint) children() (dragon.vm.torch.nn.Module method) chunk() (dragon.vm.torch.Tensor method) (in module dragon.vm.torch) clamp() (dragon.vm.torch.Tensor method) (in module dragon.vm.torch) clamp_() (dragon.vm.torch.Tensor method) clear() (dragon.Workspace method) clip() (in module dragon.math) clip_by_value() (in module dragon.vm.tensorflow) CoinFlip (class in dragon.vm.dali.ops) ColorSpaceConversion (class in dragon.vm.dali.ops) ColorTwist (class in dragon.vm.dali.ops) concat() (in module dragon) (in module dragon.vm.tensorflow) Concatenate (class in dragon.vm.tensorflow.keras.layers) Constant (class in dragon.vm.tensorflow.keras.initializers) constant() (in module dragon) (in module dragon.vm.tensorflow) constant_() (in module dragon.vm.torch.nn.init) ConstantPad1d (class in dragon.vm.torch.nn) ConstantPad2d (class in dragon.vm.torch.nn) ConstantPad3d (class in dragon.vm.torch.nn) contiguous() (dragon.vm.torch.Tensor method) Contrast (class in dragon.vm.dali.ops) conv() (in module dragon.nn) Conv1D (class in dragon.vm.tensorflow.keras.layers) Conv1d (class in dragon.vm.torch.nn) conv1d() (in module dragon.nn) (in module dragon.vm.tensorflow.nn) (in module dragon.vm.torch.nn.functional) conv1d_transpose() (in module dragon.nn) (in module dragon.vm.tensorflow.nn) Conv1DTranspose (class in dragon.vm.tensorflow.keras.layers) Conv2D (class in dragon.vm.tensorflow.keras.layers) Conv2d (class in dragon.vm.torch.nn) conv2d() (in module dragon.nn) (in module dragon.vm.tensorflow.nn) (in module dragon.vm.torch.nn.functional) conv2d_transpose() (in module dragon.nn) (in module dragon.vm.tensorflow.nn) Conv2DTranspose (class in dragon.vm.tensorflow.keras.layers) Conv3D (class in dragon.vm.tensorflow.keras.layers) Conv3d (class in dragon.vm.torch.nn) conv3d() (in module dragon.nn) (in module dragon.vm.tensorflow.nn) (in module dragon.vm.torch.nn.functional) conv3d_transpose() (in module dragon.nn) (in module dragon.vm.tensorflow.nn) Conv3DTranspose (class in dragon.vm.tensorflow.keras.layers) conv_transpose() (in module dragon.nn) (in module dragon.vm.tensorflow.nn) conv_transpose1d() (in module dragon.vm.torch.nn.functional) conv_transpose2d() (in module dragon.vm.torch.nn.functional) conv_transpose3d() (in module dragon.vm.torch.nn.functional) convert_sync_batchnorm() (dragon.vm.torch.nn.SyncBatchNorm class method) convert_to_tensor() (in module dragon.vm.tensorflow) convolution() (in module dragon.vm.tensorflow.nn) ConvTranspose1d (class in dragon.vm.torch.nn) ConvTranspose2d (class in dragon.vm.torch.nn) ConvTranspose3d (class in dragon.vm.torch.nn) copy() (dragon.DeviceSpec method) (dragon.Tensor method) (dragon.vm.torch.device method) copy_() (dragon.vm.torch.Tensor method) cos() (dragon.vm.torch.Tensor method) (in module dragon.math) (in module dragon.vm.tensorflow.math) (in module dragon.vm.torch) cosine_similarity() (in module dragon.vm.torch.nn.functional) CosineSimilarity (class in dragon.vm.torch.nn) cpu() (dragon.vm.torch.nn.Module method) (dragon.vm.torch.Tensor method) CropMirrorNormalize (class in dragon.vm.dali.ops) cross_entropy() (in module dragon.vm.torch.nn.functional) CrossEntropyLoss (class in dragon.vm.torch.nn) ctc_loss() (in module dragon.losses) (in module dragon.vm.torch.nn.functional) CTCLoss (class in dragon.vm.torch.nn) cuda() (dragon.vm.torch.nn.Module method) (dragon.vm.torch.Tensor method) cuda_engine (dragon.vm.tensorrt.Engine attribute) cumsum() (dragon.vm.torch.Tensor method) (in module dragon.math) (in module dragon.vm.tensorflow.math) (in module dragon.vm.torch) current_device() (in module dragon.cuda) (in module dragon.mps) (in module dragon.vm.torch.cuda) D data (dragon.vm.torch.Tensor attribute) debug() (in module dragon.logging) define_graph() (dragon.vm.dali.Pipeline method) Dense (class in dragon.vm.tensorflow.keras.layers) depth_to_space() (in module dragon.nn) (in module dragon.vm.tensorflow.nn) depthwise_conv2d() (in module dragon.nn) (in module dragon.vm.tensorflow.nn) (in module dragon.vm.torch.nn.functional) DepthwiseConv2D (class in dragon.vm.tensorflow.keras.layers) DepthwiseConv2d (class in dragon.vm.torch.nn) detach() (dragon.vm.torch.Tensor method) device (class in dragon.vm.torch) (dragon.Tensor attribute) (dragon.vm.torch.Tensor attribute) device() (in module dragon) (in module dragon.vm.dali) (in module dragon.vm.tensorflow) device_buffer (dragon.vm.tensorrt.Binding attribute) device_dlpack (dragon.vm.tensorrt.Binding attribute) device_id (dragon.vm.dali.Pipeline attribute) (dragon.vm.tensorrt.Binding attribute) DeviceSpec (class in dragon) dim() (dragon.vm.torch.Tensor method) dims (dragon.vm.tensorflow.TensorShape attribute) dirac_() (in module dragon.vm.torch.nn.init) div() (dragon.vm.torch.Tensor method) (in module dragon.math) (in module dragon.vm.torch) div_() (dragon.vm.torch.Tensor method) divide() (in module dragon.vm.tensorflow.math) double() (dragon.vm.torch.nn.Module method) (dragon.vm.torch.Tensor method) double_() (dragon.vm.torch.Tensor method) dragon.vm.torch.backends.cuda.matmul.allow_tf32 (built-in variable) dragon.vm.torch.backends.cudnn.allow_tf32 (built-in variable) dragon.vm.torch.backends.cudnn.benchmark (built-in variable) dragon.vm.torch.backends.cudnn.deterministic (built-in variable) dragon.vm.torch.backends.cudnn.enabled (built-in variable) drop_block() (in module dragon.nn) drop_block2d() (in module dragon.vm.torch.nn.functional) drop_path() (in module dragon.nn) (in module dragon.vm.torch.nn.functional) DropBlock2d (class in dragon.vm.torch.nn) Dropout (class in dragon.vm.tensorflow.keras.layers) (class in dragon.vm.torch.nn) dropout() (in module dragon.nn) (in module dragon.vm.tensorflow.nn) (in module dragon.vm.torch.nn.functional) DropPath (class in dragon.vm.torch.nn) DType (class in dragon.vm.tensorflow.dtypes) dtype (class in dragon.vm.torch) (dragon.Tensor attribute) (dragon.vm.tensorflow.keras.layers.Layer attribute) (dragon.vm.tensorflow.TensorSpec attribute) (dragon.vm.tensorrt.Binding attribute) (dragon.vm.torch.Tensor attribute) E eager_mode() (in module dragon) ELU (class in dragon.vm.tensorflow.keras.layers) (class in dragon.vm.torch.nn) elu() (in module dragon.nn) (in module dragon.vm.tensorflow.keras.activations) (in module dragon.vm.tensorflow.nn) (in module dragon.vm.torch.nn.functional) empty() (in module dragon.vm.torch) enable_grad (class in dragon.vm.torch) Engine (class in dragon.vm.tensorrt) engine (dragon.vm.tensorrt.onnx.BackendRep attribute) eq() (dragon.vm.torch.Tensor method) (in module dragon.vm.torch) equal() (in module dragon.math) (in module dragon.vm.tensorflow.math) Erase (class in dragon.vm.dali.ops) error() (in module dragon.logging) eval() (dragon.vm.torch.nn.Module method) exp() (dragon.vm.torch.Tensor method) (in module dragon.math) (in module dragon.vm.tensorflow.math) (in module dragon.vm.torch) exp_() (dragon.vm.torch.Tensor method) expand() (dragon.vm.torch.Tensor method) expand_as() (dragon.vm.torch.Tensor method) expand_dims() (in module dragon) (in module dragon.vm.tensorflow) exponential() (in module dragon.vm.tensorflow.keras.activations) export() (in module dragon.onnx) (in module dragon.vm.torch.onnx) extend() (dragon.vm.torch.nn.ModuleList method) ExternalSource (class in dragon.vm.dali.ops) extract_patches() (in module dragon.vision) eye() (in module dragon) (in module dragon.vm.tensorflow) (in module dragon.vm.torch) eye_() (in module dragon.vm.torch.nn.init) eye_like() (in module dragon) F fatal() (in module dragon.logging) feed_input() (dragon.vm.dali.Pipeline method) fill() (dragon.Tensor method) (in module dragon) (in module dragon.vm.tensorflow) fill_() (dragon.vm.torch.Tensor method) Flatten (class in dragon.vm.tensorflow.keras.layers) (class in dragon.vm.torch.nn) flatten() (dragon.vm.torch.Tensor method) (in module dragon) (in module dragon.vm.torch) flatten_() (dragon.vm.torch.Tensor method) flip() (dragon.vm.torch.Tensor method) (in module dragon.vm.torch) fliplr() (dragon.vm.torch.Tensor method) (in module dragon.vm.torch) flipud() (dragon.vm.torch.Tensor method) (in module dragon.vm.torch) float() (dragon.vm.torch.nn.Module method) (dragon.vm.torch.Tensor method) float_() (dragon.vm.torch.Tensor method) floor() (dragon.vm.torch.Tensor method) (in module dragon.math) (in module dragon.vm.tensorflow.math) (in module dragon.vm.torch) floor_() (dragon.vm.torch.Tensor method) forward() (dragon.vm.torch.nn.Module method) (dragon.vm.torch.nn.Sequential method) from_dlpack() (in module dragon.dlpack) (in module dragon.vm.torch.utils.dlpack) from_numpy() (in module dragon.vm.torch) full() (in module dragon.vm.torch) full_like() (in module dragon.vm.torch) function() (in module dragon) (in module dragon.vm.tensorflow) fused_batch_norm() (in module dragon.vm.tensorflow.nn) G gather() (dragon.vm.torch.Tensor method) (in module dragon) (in module dragon.vm.tensorflow) (in module dragon.vm.torch) gather_elements() (in module dragon) GaussianBlur (class in dragon.vm.dali.ops) ge() (dragon.vm.torch.Tensor method) (in module dragon.vm.torch) GELU (class in dragon.vm.torch.nn) gelu() (in module dragon.nn) (in module dragon.vm.tensorflow.nn) (in module dragon.vm.torch.nn.functional) gemm() (in module dragon.math) get() (dragon.vm.dali.Iterator method) (in module dragon.vm.tensorflow.keras.activations) (in module dragon.vm.tensorflow.keras.initializers) (in module dragon.vm.tensorflow.keras.losses) (in module dragon.vm.tensorflow.keras.regularizers) get_async() (dragon.vm.tensorrt.Binding method) get_backend() (in module dragon.distributed) get_build_info() (in module dragon.sysconfig) get_device_capability() (in module dragon.cuda) (in module dragon.vm.torch.cuda) get_device_family() (in module dragon.mps) get_device_name() (in module dragon.cuda) (in module dragon.vm.torch.cuda) get_device_type() (in module dragon.vm.dali) get_distributed_info() (in module dragon.vm.dali) get_group() (in module dragon.distributed) get_include() (in module dragon.sysconfig) get_lib() (in module dragon.sysconfig) get_num_threads() (in module dragon) get_rank() (in module dragon.distributed) get_verbosity() (in module dragon.logging) get_workspace() (in module dragon) get_world_size() (in module dragon.distributed) GlobalAveragePooling1D (class in dragon.vm.tensorflow.keras.layers) GlobalAveragePooling2D (class in dragon.vm.tensorflow.keras.layers) GlobalAveragePooling3D (class in dragon.vm.tensorflow.keras.layers) GlobalMaxPooling1D (class in dragon.vm.tensorflow.keras.layers) GlobalMaxPooling2D (class in dragon.vm.tensorflow.keras.layers) GlobalMaxPooling3D (class in dragon.vm.tensorflow.keras.layers) glorot_normal() (dragon.Tensor method) (in module dragon.random) glorot_uniform() (dragon.Tensor method) (in module dragon.random) GlorotNormal (class in dragon.vm.tensorflow.keras.initializers) GlorotUniform (class in dragon.vm.tensorflow.keras.initializers) grad (dragon.vm.torch.Tensor attribute) gradient() (dragon.GradientTape method) GradientTape (class in dragon) (class in dragon.vm.tensorflow) graph_mode() (in module dragon) greater() (in module dragon.math) (in module dragon.vm.tensorflow.math) greater_equal() (in module dragon.math) (in module dragon.vm.tensorflow.math) group_norm() (in module dragon.nn) (in module dragon.vm.torch.nn.functional) GroupNorm (class in dragon.vm.torch.nn) GRU (class in dragon.nn) (class in dragon.vm.torch.nn) gt() (dragon.vm.torch.Tensor method) (in module dragon.vm.torch) GumbelSoftmax (class in dragon.vm.torch.nn) H half() (dragon.vm.torch.nn.Module method) (dragon.vm.torch.Tensor method) half_() (dragon.vm.torch.Tensor method) handlers (dragon.vm.dali.Iterator attribute) hard_sigmoid() (in module dragon.vm.tensorflow.keras.activations) Hardsigmoid (class in dragon.vm.torch.nn) hardsigmoid() (in module dragon.nn) (in module dragon.vm.torch.nn.functional) Hardswish (class in dragon.vm.torch.nn) hardswish() (in module dragon.nn) (in module dragon.vm.torch.nn.functional) host_buffer (dragon.vm.tensorrt.Binding attribute) host_dlpack (dragon.vm.tensorrt.Binding attribute) Hsv (class in dragon.vm.dali.ops) I id (dragon.Tensor attribute) (dragon.vm.torch.Tensor attribute) Identity (class in dragon.vm.torch.nn) identity() (in module dragon) (in module dragon.vm.tensorflow) ImageDecoder (class in dragon.vm.dali.ops) ImageDecoderRandomCrop (class in dragon.vm.dali.ops) index (dragon.DeviceSpec attribute) (dragon.vm.tensorrt.Binding attribute) (dragon.vm.torch.device attribute) index_select() (dragon.vm.torch.Tensor method) (in module dragon.vm.torch) info() (in module dragon.logging) Initializer (class in dragon.vm.tensorflow.keras.initializers) Input() (in module dragon.vm.tensorflow.keras) inputs (dragon.vm.tensorrt.Engine attribute) insert() (dragon.vm.torch.nn.ModuleList method) instance_norm() (in module dragon.nn) int() (dragon.vm.torch.Tensor method) int_() (dragon.vm.torch.Tensor method) interpolate() (in module dragon.vm.torch.nn.functional) invert() (in module dragon.bitwise) (in module dragon.vm.tensorflow.bitwise) is_available() (dragon.vm.torch.backends.cudnn method) (dragon.vm.torch.backends.mps method) (dragon.vm.torch.backends.openmp method) (in module dragon.cuda) (in module dragon.mps) (in module dragon.vm.torch.cuda) is_bool (dragon.vm.tensorflow.dtypes.DType attribute) is_built() (dragon.vm.torch.backends.cuda method) (dragon.vm.torch.backends.mps method) is_compatible_with() (dragon.vm.tensorflow.dtypes.DType method) is_complex (dragon.vm.tensorflow.dtypes.DType attribute) is_contiguous() (dragon.vm.torch.Tensor method) is_finite() (in module dragon.math) (in module dragon.vm.tensorflow.math) is_floating (dragon.vm.tensorflow.dtypes.DType attribute) is_floating_point() (dragon.vm.torch.Tensor method) is_inf() (in module dragon.math) (in module dragon.vm.tensorflow.math) is_initialized() (in module dragon.distributed) is_input (dragon.vm.tensorrt.Binding attribute) is_integer (dragon.vm.tensorflow.dtypes.DType attribute) is_leaf (dragon.vm.torch.Tensor attribute) is_mpi_available() (in module dragon.distributed) is_nan() (in module dragon.math) (in module dragon.vm.tensorflow.math) is_nccl_available() (in module dragon.distributed) is_numpy_compatible (dragon.vm.tensorflow.dtypes.DType attribute) is_quantized (dragon.vm.tensorflow.dtypes.DType attribute) is_unsigned (dragon.vm.tensorflow.dtypes.DType attribute) isfinite() (dragon.vm.torch.Tensor method) (in module dragon.vm.torch) isinf() (dragon.vm.torch.Tensor method) (in module dragon.vm.torch) isnan() (dragon.vm.torch.Tensor method) (in module dragon.vm.torch) item() (dragon.Tensor method) (dragon.vm.torch.Tensor method) iterations (dragon.vm.tensorflow.keras.optimizers.Optimizer attribute) Iterator (class in dragon.vm.dali) K kaiming_normal_() (in module dragon.vm.torch.nn.init) kaiming_uniform_() (in module dragon.vm.torch.nn.init) kl_div() (in module dragon.vm.torch.nn.functional) KLDivLoss (class in dragon.vm.torch.nn) L L1 (class in dragon.vm.tensorflow.keras.regularizers) l1_l2() (in module dragon.vm.tensorflow.keras.regularizers) l1_loss() (in module dragon.losses) (in module dragon.vm.torch.nn.functional) L1L2 (class in dragon.vm.tensorflow.keras.regularizers) L1Loss (class in dragon.vm.torch.nn) L2 (class in dragon.vm.tensorflow.keras.regularizers) l2_loss() (in module dragon.losses) (in module dragon.vm.tensorflow.nn) l2_normalize() (in module dragon.vm.tensorflow.math) LARS (class in dragon.vm.torch.optim) Layer (class in dragon.vm.tensorflow.keras.layers) layer_norm() (in module dragon.nn) (in module dragon.vm.torch.nn.functional) LayerNorm (class in dragon.vm.torch.nn) LayerNormalization (class in dragon.vm.tensorflow.keras.layers) layers (dragon.vm.tensorflow.keras.Sequential attribute) le() (dragon.vm.torch.Tensor method) (in module dragon.vm.torch) leaky_relu() (in module dragon.nn) (in module dragon.vm.tensorflow.nn) (in module dragon.vm.torch.nn.functional) LeakyReLU (class in dragon.vm.tensorflow.keras.layers) (class in dragon.vm.torch.nn) less() (in module dragon.math) (in module dragon.vm.tensorflow.math) less_equal() (in module dragon.math) (in module dragon.vm.tensorflow.math) limits (dragon.vm.tensorflow.dtypes.DType attribute) Linear (class in dragon.vm.torch.nn) linear() (in module dragon.vm.tensorflow.keras.activations) (in module dragon.vm.torch.nn.functional) linspace() (in module dragon) (in module dragon.vm.tensorflow) (in module dragon.vm.torch) load_library() (in module dragon) load_state_dict() (dragon.vm.torch.nn.Module method) load_weights() (dragon.vm.tensorflow.keras.layers.Layer method) local_response_norm() (in module dragon.nn) (in module dragon.vm.torch.nn.functional) local_response_normalization() (in module dragon.vm.tensorflow.nn) LocalResponseNorm (class in dragon.vm.torch.nn) log() (dragon.vm.torch.Tensor method) (in module dragon.logging) (in module dragon.math) (in module dragon.vm.tensorflow.math) (in module dragon.vm.torch) log_() (dragon.vm.torch.Tensor method) log_softmax() (in module dragon.nn) (in module dragon.vm.tensorflow.nn) (in module dragon.vm.torch.nn.functional) logical_and() (dragon.vm.torch.Tensor method) (in module dragon.math) (in module dragon.vm.torch) logical_not() (dragon.vm.torch.Tensor method) (in module dragon.math) (in module dragon.vm.torch) logical_or() (dragon.vm.torch.Tensor method) (in module dragon.math) (in module dragon.vm.torch) logical_xor() (dragon.vm.torch.Tensor method) (in module dragon.math) (in module dragon.vm.torch) LogSoftmax (class in dragon.vm.torch.nn) logsumexp() (dragon.vm.torch.Tensor method) (in module dragon.vm.torch) long() (dragon.vm.torch.Tensor method) long_() (dragon.vm.torch.Tensor method) Loss (class in dragon.vm.tensorflow.keras.losses) lp_norm() (in module dragon.nn) LSTM (class in dragon.nn) (class in dragon.vm.torch.nn) LSTMCell (class in dragon.vm.torch.nn) lt() (dragon.vm.torch.Tensor method) (in module dragon.vm.torch) M masked_fill_() (dragon.vm.torch.Tensor method) masked_select() (dragon.vm.torch.Tensor method) (in module dragon.vm.torch) matmul() (dragon.vm.torch.Tensor method) (in module dragon.math) (in module dragon.vm.tensorflow.linalg) (in module dragon.vm.torch) max (dragon.vm.tensorflow.dtypes.DType attribute) max() (dragon.vm.torch.Tensor method) (in module dragon.math) (in module dragon.vm.torch) max_batch_size (dragon.vm.dali.Pipeline attribute) max_pool() (in module dragon.vm.tensorflow.nn) max_pool1d() (in module dragon.vm.tensorflow.nn) (in module dragon.vm.torch.nn.functional) max_pool2d() (in module dragon.vm.tensorflow.nn) (in module dragon.vm.torch.nn.functional) max_pool3d() (in module dragon.vm.tensorflow.nn) (in module dragon.vm.torch.nn.functional) Maximum (class in dragon.vm.tensorflow.keras.layers) maximum() (dragon.vm.torch.Tensor method) (in module dragon.math) (in module dragon.vm.torch) MaxPool1d (class in dragon.vm.torch.nn) MaxPool2d (class in dragon.vm.torch.nn) MaxPool3d (class in dragon.vm.torch.nn) MaxPooling1D (class in dragon.vm.tensorflow.keras.layers) MaxPooling2D (class in dragon.vm.tensorflow.keras.layers) MaxPooling3D (class in dragon.vm.tensorflow.keras.layers) mean() (dragon.vm.torch.Tensor method) (in module dragon.math) (in module dragon.vm.torch) mean_absolute_error() (in module dragon.vm.tensorflow.keras.losses) mean_squared_error() (in module dragon.vm.tensorflow.keras.losses) MeanAbsoluteError (class in dragon.vm.tensorflow.keras.losses) MeanSquaredError (class in dragon.vm.tensorflow.keras.losses) memory_allocated() (dragon.Workspace method) (in module dragon.cuda) merge_from() (dragon.Workspace method) min (dragon.vm.tensorflow.dtypes.DType attribute) min() (dragon.vm.torch.Tensor method) (in module dragon.math) (in module dragon.vm.torch) Minimum (class in dragon.vm.tensorflow.keras.layers) minimum() (dragon.vm.torch.Tensor method) (in module dragon.math) (in module dragon.vm.torch) mm() (dragon.vm.torch.Tensor method) (in module dragon.vm.torch) Module (class in dragon.vm.tensorflow) (class in dragon.vm.torch.nn) ModuleList (class in dragon.vm.torch.nn) modules() (dragon.vm.torch.nn.Module method) moments() (in module dragon.nn) (in module dragon.vm.tensorflow.nn) mse_loss() (in module dragon.vm.torch.nn.functional) MSELoss (class in dragon.vm.torch.nn) mul() (dragon.vm.torch.Tensor method) (in module dragon.math) (in module dragon.vm.torch) mul_() (dragon.vm.torch.Tensor method) multi_head_attention_forward() (in module dragon.vm.torch.nn.functional) MultiheadAttention (class in dragon.vm.torch.nn) multinomial() (dragon.vm.torch.Tensor method) (in module dragon.random) (in module dragon.vm.torch) Multiply (class in dragon.vm.tensorflow.keras.layers) multiply() (in module dragon.vm.tensorflow.math) N name (dragon.Tensor attribute) (dragon.vm.tensorflow.dtypes.DType attribute) (dragon.vm.tensorflow.Module attribute) (dragon.vm.tensorflow.TensorSpec attribute) (dragon.vm.tensorrt.Binding attribute) name_scope (dragon.vm.tensorflow.Module attribute) name_scope() (in module dragon) (in module dragon.vm.tensorflow) named_buffers() (dragon.vm.torch.nn.Module method) named_children() (dragon.vm.torch.nn.Module method) named_modules() (dragon.vm.torch.nn.Module method) named_parameters() (dragon.vm.torch.nn.Module method) narrow() (dragon.vm.torch.Tensor method) (in module dragon.vm.torch) ndim (dragon.Tensor attribute) ndimension() (dragon.vm.torch.Tensor method) ndims (dragon.vm.tensorflow.TensorShape attribute) ne() (dragon.vm.torch.Tensor method) (in module dragon.vm.torch) neg() (dragon.vm.torch.Tensor method) (in module dragon.vm.torch) neg_() (dragon.vm.torch.Tensor method) negative() (in module dragon.math) (in module dragon.vm.tensorflow.math) new_empty() (dragon.vm.torch.Tensor method) new_full() (dragon.vm.torch.Tensor method) new_group() (in module dragon.distributed) new_ones() (dragon.vm.torch.Tensor method) new_tensor() (dragon.vm.torch.Tensor method) new_zeros() (dragon.vm.torch.Tensor method) next() (dragon.vm.dali.Iterator method) nll_loss() (in module dragon.losses) (in module dragon.vm.torch.nn.functional) NLLLoss (class in dragon.vm.torch.nn) no_checkpoint (class in dragon.vm.torch.utils.checkpoint) no_grad (class in dragon.vm.torch) non_trainable_weights (dragon.vm.tensorflow.keras.layers.Layer attribute) nonzero() (dragon.vm.torch.Tensor method) (in module dragon) (in module dragon.vm.torch) norm() (dragon.vm.torch.Tensor method) (in module dragon.math) (in module dragon.vm.torch) normal() (dragon.Tensor method) (in module dragon.random) (in module dragon.vm.tensorflow.random) normal_() (dragon.vm.torch.Tensor method) (in module dragon.vm.torch.nn.init) normal_like() (in module dragon.random) Normalize (class in dragon.vm.dali.ops) normalize() (in module dragon.vm.torch.nn.functional) not_equal() (in module dragon.math) (in module dragon.vm.tensorflow.math) num_threads (dragon.vm.dali.Pipeline attribute) numel() (dragon.vm.torch.Size method) (dragon.vm.torch.Tensor method) numpy() (dragon.Tensor method) (dragon.vm.torch.Tensor method) O one_() (dragon.vm.torch.Tensor method) one_hot() (in module dragon) (in module dragon.vm.tensorflow) (in module dragon.vm.torch.nn.functional) Ones (class in dragon.vm.tensorflow.keras.initializers) ones() (in module dragon) (in module dragon.vm.tensorflow) (in module dragon.vm.torch) ones_() (in module dragon.vm.torch.nn.init) ones_like() (in module dragon) (in module dragon.vm.tensorflow) (in module dragon.vm.torch) Optimizer (class in dragon.optimizers) (class in dragon.vm.tensorflow.keras.optimizers) (class in dragon.vm.torch.optim) outputs (dragon.vm.tensorrt.Engine attribute) P Pad (class in dragon.vm.dali.ops) pad() (in module dragon) (in module dragon.vm.tensorflow) (in module dragon.vm.torch.nn.functional) Parameter (class in dragon.vm.torch.nn) parameters() (dragon.vm.torch.nn.Module method) Paste (class in dragon.vm.dali.ops) permutation() (in module dragon.random) Permute (class in dragon.vm.tensorflow.keras.layers) permute() (dragon.vm.torch.Tensor method) (in module dragon.vm.torch) permute_() (dragon.vm.torch.Tensor method) Pipeline (class in dragon.vm.dali) pixel_shuffle() (in module dragon.vm.torch.nn.functional) pixel_unshuffle() (in module dragon.vm.torch.nn.functional) PixelShuffle (class in dragon.vm.torch.nn) PixelUnshuffle (class in dragon.vm.torch.nn) pool() (in module dragon.nn) pool1d() (in module dragon.nn) pool2d() (in module dragon.nn) pool3d() (in module dragon.nn) pop() (dragon.vm.tensorflow.keras.Sequential method) pow() (dragon.vm.torch.Tensor method) (in module dragon.math) (in module dragon.vm.tensorflow.math) (in module dragon.vm.torch) PReLU (class in dragon.vm.torch.nn) prelu() (in module dragon.nn) (in module dragon.vm.torch.nn.functional) prepare_backend() (in module dragon.onnx) (in module dragon.vm.tensorrt.onnx) ptr (dragon.cuda.Stream attribute) python_plugin() (in module dragon) R rand() (in module dragon.vm.torch) randn() (in module dragon.vm.torch) RandomBBoxCrop (class in dragon.vm.dali.ops) RandomNormal (class in dragon.vm.tensorflow.keras.initializers) RandomResizedCrop (class in dragon.vm.dali.ops) RandomUniform (class in dragon.vm.tensorflow.keras.initializers) randperm() (in module dragon.vm.torch) range() (in module dragon) (in module dragon.vm.tensorflow) rank (dragon.vm.tensorflow.TensorShape attribute) real_dtype (dragon.vm.tensorflow.dtypes.DType attribute) reciprocal() (dragon.vm.torch.Tensor method) (in module dragon.math) (in module dragon.vm.tensorflow.math) (in module dragon.vm.torch) reciprocal_() (dragon.vm.torch.Tensor method) record() (in module dragon.onnx) reduce_max() (in module dragon.vm.tensorflow.math) reduce_mean() (in module dragon.vm.tensorflow.math) reduce_min() (in module dragon.vm.tensorflow.math) reduce_sum() (in module dragon.vm.tensorflow.math) reduce_variance() (in module dragon.vm.tensorflow.math) ReflectionPad1d (class in dragon.vm.torch.nn) ReflectionPad2d (class in dragon.vm.torch.nn) ReflectionPad3d (class in dragon.vm.torch.nn) register_buffer() (dragon.vm.torch.nn.Module method) register_forward_hook() (dragon.vm.torch.nn.Module method) register_parameter() (dragon.vm.torch.nn.Module method) Regularizer (class in dragon.vm.tensorflow.keras.regularizers) ReLU (class in dragon.vm.tensorflow.keras.layers) (class in dragon.vm.torch.nn) relu() (in module dragon.nn) (in module dragon.vm.tensorflow.keras.activations) (in module dragon.vm.tensorflow.nn) (in module dragon.vm.torch.nn.functional) ReLU6 (class in dragon.vm.torch.nn) relu6() (in module dragon.nn) (in module dragon.vm.tensorflow.nn) (in module dragon.vm.torch.nn.functional) repeat() (dragon.vm.torch.Tensor method) (in module dragon) ReplicationPad1d (class in dragon.vm.torch.nn) ReplicationPad2d (class in dragon.vm.torch.nn) ReplicationPad3d (class in dragon.vm.torch.nn) requires_grad (dragon.Tensor attribute) (dragon.vm.torch.Tensor attribute) reset() (dragon.GradientTape method) reset_workspace() (in module dragon) Reshape (class in dragon.vm.dali.ops) (class in dragon.vm.tensorflow.keras.layers) reshape() (dragon.Tensor method) (dragon.vm.torch.Tensor method) (in module dragon) (in module dragon.vm.tensorflow) (in module dragon.vm.torch) reshape_() (dragon.vm.torch.Tensor method) Resize (class in dragon.vm.dali.ops) resize() (in module dragon.vision) retain_grad() (dragon.vm.torch.Tensor method) reverse() (in module dragon) (in module dragon.vm.tensorflow) RMSprop (class in dragon.optimizers) (class in dragon.vm.tensorflow.keras.optimizers) (class in dragon.vm.torch.optim) RNN (class in dragon.nn) (class in dragon.vm.torch.nn) roi_align() (in module dragon.vision) (in module dragon.vm.torchvision.ops) roi_pool() (in module dragon.vision) (in module dragon.vm.torchvision.ops) roll() (dragon.vm.torch.Tensor method) (in module dragon) (in module dragon.vm.tensorflow) (in module dragon.vm.torch) Rotate (class in dragon.vm.dali.ops) round() (dragon.vm.torch.Tensor method) (in module dragon.math) (in module dragon.vm.tensorflow.math) (in module dragon.vm.torch) round_() (dragon.vm.torch.Tensor method) rsqrt() (dragon.vm.torch.Tensor method) (in module dragon.math) (in module dragon.vm.tensorflow.math) (in module dragon.vm.torch) rsqrt_() (dragon.vm.torch.Tensor method) run() (dragon.onnx.BackendRep method) (dragon.vm.tensorrt.Engine method) (dragon.vm.tensorrt.onnx.BackendRep method) run_model() (in module dragon.onnx) (in module dragon.vm.tensorrt.onnx) run_node() (in module dragon.vm.tensorrt.onnx) S save_weights() (dragon.vm.tensorflow.keras.layers.Layer method) scatter() (dragon.vm.torch.Tensor method) (in module dragon.vm.torch) scatter_() (dragon.vm.torch.Tensor method) scatter_add() (dragon.vm.torch.Tensor method) (in module dragon) (in module dragon.vm.torch) scatter_add_() (dragon.vm.torch.Tensor method) scatter_elements() (in module dragon) SELU (class in dragon.vm.tensorflow.keras.layers) (class in dragon.vm.torch.nn) selu() (in module dragon.nn) (in module dragon.vm.tensorflow.keras.activations) (in module dragon.vm.tensorflow.nn) (in module dragon.vm.torch.nn.functional) Sequential (class in dragon.vm.tensorflow.keras) (class in dragon.vm.torch.nn) set_cublas_flags() (in module dragon.cuda) set_cudnn_flags() (in module dragon.cuda) set_default_device() (in module dragon.cuda) (in module dragon.mps) set_device() (in module dragon.cuda) (in module dragon.mps) (in module dragon.vm.torch.cuda) set_directory() (in module dragon.logging) set_grad_enabled (class in dragon.vm.torch) set_num_threads() (in module dragon) set_optimization() (in module dragon.autograph) set_scheduler() (in module dragon.autograph) set_seed() (in module dragon.random) set_verbosity() (in module dragon.autograph) (in module dragon.logging) SGD (class in dragon.optimizers) (class in dragon.vm.tensorflow.keras.optimizers) (class in dragon.vm.torch.optim) shape (dragon.Tensor attribute) (dragon.vm.tensorflow.TensorSpec attribute) (dragon.vm.tensorrt.Binding attribute) (dragon.vm.torch.Tensor attribute) shape() (in module dragon) (in module dragon.vm.tensorflow) Sigmoid (class in dragon.vm.torch.nn) sigmoid() (in module dragon.math) (in module dragon.vm.tensorflow.keras.activations) (in module dragon.vm.tensorflow.math) (in module dragon.vm.torch.nn.functional) sigmoid_cross_entropy_loss() (in module dragon.losses) sigmoid_cross_entropy_with_logits() (in module dragon.vm.tensorflow.nn) sigmoid_focal_loss() (in module dragon.losses) (in module dragon.vm.torch.nn.functional) SigmoidFocalLoss (class in dragon.vm.torch.nn) sign() (dragon.vm.torch.Tensor method) (in module dragon.math) (in module dragon.vm.tensorflow.math) (in module dragon.vm.torch) sign_() (dragon.vm.torch.Tensor method) SiLU (class in dragon.vm.torch.nn) silu() (in module dragon.nn) (in module dragon.vm.tensorflow.nn) (in module dragon.vm.torch.nn.functional) sin() (dragon.vm.torch.Tensor method) (in module dragon.math) (in module dragon.vm.tensorflow.math) (in module dragon.vm.torch) Size (class in dragon.vm.torch) size (dragon.Tensor attribute) size() (dragon.vm.torch.Tensor method) Slice (class in dragon.vm.dali.ops) slice() (in module dragon) (in module dragon.vm.tensorflow) smooth_l1_loss() (in module dragon.losses) (in module dragon.vm.torch.nn.functional) SmoothL1Loss (class in dragon.vm.torch.nn) Softmax (class in dragon.vm.tensorflow.keras.layers) (class in dragon.vm.torch.nn) softmax() (in module dragon.nn) (in module dragon.vm.tensorflow.keras.activations) (in module dragon.vm.tensorflow.nn) (in module dragon.vm.torch.nn.functional) softmax_cross_entropy_loss() (in module dragon.losses) softmax_cross_entropy_with_logits() (in module dragon.vm.tensorflow.nn) sort() (dragon.vm.torch.Tensor method) (in module dragon) (in module dragon.vm.tensorflow) (in module dragon.vm.torch) space_to_depth() (in module dragon.nn) (in module dragon.vm.tensorflow.nn) sparse_categorical_crossentropy() (in module dragon.vm.tensorflow.keras.losses) sparse_softmax_cross_entropy_with_logits() (in module dragon.vm.tensorflow.nn) SparseCategoricalCrossentropy (class in dragon.vm.tensorflow.keras.losses) split() (dragon.vm.torch.Tensor method) (in module dragon) (in module dragon.vm.tensorflow) (in module dragon.vm.torch) sqrt() (dragon.vm.torch.Tensor method) (in module dragon.math) (in module dragon.vm.tensorflow.math) (in module dragon.vm.torch) sqrt_() (dragon.vm.torch.Tensor method) square() (dragon.vm.torch.Tensor method) (in module dragon.math) (in module dragon.vm.tensorflow.math) (in module dragon.vm.torch) squeeze() (dragon.vm.torch.Tensor method) (in module dragon) (in module dragon.vm.tensorflow) (in module dragon.vm.torch) squeeze_() (dragon.vm.torch.Tensor method) stack() (in module dragon) (in module dragon.vm.torch) state_dict() (dragon.vm.torch.nn.Module method) step() (dragon.vm.torch.optim.Optimizer method) stop_gradient() (in module dragon) stop_recording() (dragon.GradientTape method) Stream (class in dragon.cuda) sub() (dragon.vm.torch.Tensor method) (in module dragon.math) (in module dragon.vm.torch) sub_() (dragon.vm.torch.Tensor method) submodules (dragon.vm.tensorflow.Module attribute) Subtract (class in dragon.vm.tensorflow.keras.layers) subtract() (in module dragon.vm.tensorflow.math) sum() (dragon.vm.torch.Tensor method) (in module dragon.math) (in module dragon.vm.torch) sum_grad() (dragon.vm.torch.optim.Optimizer method) supports_device() (in module dragon.onnx) (in module dragon.vm.tensorrt.onnx) swish() (in module dragon.vm.tensorflow.keras.activations) sync_batch_norm() (in module dragon.nn) (in module dragon.vm.torch.nn.functional) SyncBatchNorm (class in dragon.vm.torch.nn) synchronize() (dragon.cuda.Stream method) (in module dragon.cuda) (in module dragon.mps) (in module dragon.vm.torch.cuda) T T (dragon.Tensor attribute) (dragon.vm.torch.Tensor attribute) Tanh (class in dragon.vm.torch.nn) tanh() (in module dragon.math) (in module dragon.vm.tensorflow.keras.activations) (in module dragon.vm.tensorflow.math) (in module dragon.vm.torch.nn.functional) Tensor (class in dragon) (class in dragon.vm.torch) tensor() (in module dragon.vm.torch) TensorShape (class in dragon.vm.tensorflow) TensorSpec (class in dragon.vm.tensorflow) TFRecordReader (class in dragon.vm.dali.ops) tile() (in module dragon) (in module dragon.vm.tensorflow) (in module dragon.vm.torch) to() (dragon.vm.torch.nn.Module method) (dragon.vm.torch.Tensor method) to_dlpack() (in module dragon.dlpack) (in module dragon.vm.torch.utils.dlpack) tolist() (dragon.Tensor method) (dragon.vm.torch.Tensor method) top_k() (in module dragon.math) (in module dragon.vm.tensorflow.math) topk() (dragon.vm.torch.Tensor method) (in module dragon.vm.torch) trace() (in module dragon.vm.torch.jit) train() (dragon.vm.torch.nn.Module method) trainable (dragon.vm.tensorflow.keras.layers.Layer attribute) trainable_variables (dragon.vm.tensorflow.Module attribute) trainable_weights (dragon.vm.tensorflow.keras.layers.Layer attribute) TransformerDecoder (class in dragon.vm.torch.nn) TransformerDecoderLayer (class in dragon.vm.torch.nn) TransformerEncoder (class in dragon.vm.torch.nn) TransformerEncoderLayer (class in dragon.vm.torch.nn) transpose() (dragon.Tensor method) (dragon.vm.torch.Tensor method) (in module dragon) (in module dragon.vm.tensorflow) (in module dragon.vm.torch) transpose_() (dragon.vm.torch.Tensor method) tril() (dragon.vm.torch.Tensor method) (in module dragon) (in module dragon.vm.torch) tril_() (dragon.vm.torch.Tensor method) triu() (dragon.vm.torch.Tensor method) (in module dragon) (in module dragon.vm.torch) triu_() (dragon.vm.torch.Tensor method) trunc_normal_() (in module dragon.vm.torch.nn.init) truncated_normal() (dragon.Tensor method) (in module dragon.random) (in module dragon.vm.tensorflow.random) TruncatedNormal (class in dragon.vm.tensorflow.keras.initializers) type (dragon.DeviceSpec attribute) (dragon.vm.torch.device attribute) type() (dragon.vm.torch.Tensor method) U unbind() (dragon.vm.torch.Tensor method) (in module dragon.vm.torch) Unfold (class in dragon.vm.torch.nn) unfold() (in module dragon.vm.torch.nn.functional) Uniform (class in dragon.vm.dali.ops) uniform() (dragon.Tensor method) (in module dragon.random) (in module dragon.vm.tensorflow.random) uniform_() (dragon.vm.torch.Tensor method) (in module dragon.vm.torch.nn.init) uniform_like() (in module dragon.random) unique() (dragon.vm.torch.Tensor method) (in module dragon) (in module dragon.vm.tensorflow) (in module dragon.vm.torch) unique_with_counts() (in module dragon.vm.tensorflow) unsqueeze() (dragon.vm.torch.Tensor method) (in module dragon.vm.torch) unsqueeze_() (dragon.vm.torch.Tensor method) unstack() (in module dragon) (in module dragon.vm.tensorflow) Upsample (class in dragon.vm.torch.nn) upsample() (in module dragon.vm.torch.nn.functional) upsample_bilinear() (in module dragon.vm.torch.nn.functional) upsample_nearest() (in module dragon.vm.torch.nn.functional) UpSampling1D (class in dragon.vm.tensorflow.keras.layers) UpSampling2D (class in dragon.vm.tensorflow.keras.layers) UpSampling3D (class in dragon.vm.tensorflow.keras.layers) UpsamplingBilinear2d (class in dragon.vm.torch.nn) UpsamplingNearest2d (class in dragon.vm.torch.nn) V var() (dragon.vm.torch.Tensor method) (in module dragon.math) (in module dragon.vm.torch) var_mean() (in module dragon.vm.torch) variable_scope() (in module dragon) variables (dragon.vm.tensorflow.Module attribute) VarianceScaling (class in dragon.vm.tensorflow.keras.initializers) version() (dragon.vm.torch.backends.cudnn method) view() (dragon.vm.torch.Tensor method) view_() (dragon.vm.torch.Tensor method) view_as() (dragon.vm.torch.Tensor method) W warning() (in module dragon.logging) WarpAffine (class in dragon.vm.dali.ops) watch() (dragon.GradientTape method) weights (dragon.vm.tensorflow.keras.layers.Layer attribute) where() (dragon.vm.torch.Tensor method) (in module dragon) (in module dragon.vm.torch) Workspace (class in dragon) X xavier_normal_() (in module dragon.vm.torch.nn.init) xavier_uniform_() (in module dragon.vm.torch.nn.init) Z zero_() (dragon.vm.torch.Tensor method) zero_grad() (dragon.vm.torch.nn.Module method) (dragon.vm.torch.optim.Optimizer method) ZeroPad2d (class in dragon.vm.torch.nn) ZeroPadding1D (class in dragon.vm.tensorflow.keras.layers) ZeroPadding2D (class in dragon.vm.tensorflow.keras.layers) ZeroPadding3D (class in dragon.vm.tensorflow.keras.layers) Zeros (class in dragon.vm.tensorflow.keras.initializers) zeros() (in module dragon) (in module dragon.vm.tensorflow) (in module dragon.vm.torch) zeros_() (in module dragon.vm.torch.nn.init) zeros_like() (in module dragon) (in module dragon.vm.tensorflow) (in module dragon.vm.torch)
