欢迎查阅计图文档
模块API:
- jittor
- jittor.nn
BCELossBCEWithLogitsLossBatchNormBatchNorm1dBatchNorm2dBatchNorm3dBilinearConstantPad2dConvConv1dConv1d_spConv2dConvTransposeConvTranspose2dConvTranspose3dCrossEntropyLossDropPathDropoutDropout2dELUEmbeddingFlattenGRUGRUCellGroupNormIdentityInstanceNormInstanceNorm1dInstanceNorm2dInstanceNorm3dKLDivLossL1LossLSTMLSTMCellLayerNormLayerNorm1dLayerNorm2dLayerNorm3dLinearMSELossMishModuleListPReLUParameter()ParameterDictPixelShuffleRNNRNNBaseRNNCellReflectionPad2dReplicationPad2dResizeSequentialSigmoidSoftplusTanhUpsampleUpsamplingBilinear2dUpsamplingNearest2dZeroPad2daffine_grid()affine_grid_generator_4D()affine_grid_generator_5D()backward()baddbmm()batch_norm()bce_loss()bilinear()binary_cross_entropy_with_logits()bmm()bmm_transpose()clip_coordinates()conv()conv2d()conv3d()conv_transpose()conv_transpose2d()conv_transpose3d()cross_entropy_loss()dropout()dropout2d()droppath()elu()embedding()fold()fp32_guard()gelu()get_init_var_rand()grid_sample()grid_sample_v0()grid_sampler()grid_sampler_2d()grid_sampler_3d()grid_sampler_compute_source_index()grid_sampler_unnormalize()group_norm()hardtanh()identity()instance_norm()interpolate()l1_loss()layer_norm()leaky_relu()linear()linspace_from_neg_one()log_sigmoid()log_softmax()logsumexp()make_base_grid_4D()make_base_grid_5D()matmul()matmul_transpose()mish()mse_loss()nll_loss()one_hot()pad()reflect_coordinates()relu()relu6()resize()sign()silu()skip_init()smooth_l1_loss()softmax()softplus()unfold()upsample()AdaptiveAvgPool2dAdaptiveAvgPool3dAdaptiveMaxPool2dAdaptiveMaxPool2dAvgPool2dAvgPool3dMaxPool2dMaxPool3dMaxUnpool2dMaxUnpool3dPoolPool3davg_pool2d()max_pool2d()max_pool3d()pool()pool3d()ComplexNumber
- jittor.models
AlexNetDenseNetGoogLeNetInception3MNASNetMobileNetV2ResNetResnet101()Resnet152()Resnet18()Resnet26()Resnet34()Resnet38()Resnet50()Resnext101_32x8d()Resnext50_32x4d()ShuffleNetV2SqueezeNetVGGWide_resnet101_2()Wide_resnet50_2()alexnet()densenet121()densenet161()densenet169()densenet201()googlenet()inception_v3()mnasnet0_5()mnasnet0_75()mnasnet1_0()mnasnet1_3()mobilenet_v2()res2net101()res2net50()resnet101()resnet152()resnet18()resnet26()resnet34()resnet38()resnet50()resnext101_32x8d()resnext50_32x4d()shufflenet_v2_x0_5()shufflenet_v2_x1_0()shufflenet_v2_x1_5()shufflenet_v2_x2_0()squeezenet1_0()squeezenet1_1()vgg11()vgg11_bn()vgg13()vgg13_bn()vgg16()vgg16_bn()vgg19()vgg19_bn()wide_resnet101_2()wide_resnet50_2()
- jittor.optim
- jittor.init
calculate_gain()calculate_std()constant()constant_()eye()eye_()fill()gauss()gauss_()invariant_uniform()invariant_uniform_()kaiming_normal_()kaiming_uniform_()one()one_()random_()relu_invariant_gauss()relu_invariant_gauss_()trunc_normal_()uniform()uniform_()xavier_gauss()xavier_gauss_()xavier_uniform()xavier_uniform_()zero()zero_()
- jittor.contrib
- jittor.dataset
- jittor.transform
CenterCropColorJitterComposeCropFiveCropGrayImageNormalizeLambdaRandomAffineRandomApplyRandomChoiceRandomCropRandomCropAndResizeRandomGrayRandomHorizontalFlipRandomOrderRandomPerspectiveRandomResizedCropRandomRotationRandomSizedCropRandomVerticalFlipResizeTenCropToPILImageToTensoradjust_brightness()adjust_contrast()adjust_gamma()adjust_hue()adjust_saturation()center_crop()crop()crop_and_resize()gray()hflip()image_normalize()pil_to_tensor()resize()to_pil_image()to_tensor()vflip()
- jittor.mpi
- jittor.linalg
- jittor.console
- jittor.distributions
- jittor.attention
- jittor.einops
- jittor.sparse
- jittor.weightnorm