Module: tfa | TensorFlow Addons

Useful extra functionality for TensorFlow maintained by SIG-addons. Install Learn Introduction NewtoTensorFlow? TensorFlow ThecoreopensourceMLlibrary ForJavaScript TensorFlow.jsforMLusingJavaScript ForMobile&Edge TensorFlowLiteformobileandedgedevices ForProduction TensorFlowExtendedforend-to-endMLcomponents API TensorFlow(v2.10.0) Versions… TensorFlow.js TensorFlowLite TFX Resources Models&datasets Pre-trainedmodelsanddatasetsbuiltbyGoogleandthecommunity Tools EcosystemoftoolstohelpyouuseTensorFlow Libraries&extensions LibrariesandextensionsbuiltonTensorFlow TensorFlowCertificateprogram DifferentiateyourselfbydemonstratingyourMLproficiency LearnML EducationalresourcestolearnthefundamentalsofMLwithTensorFlow ResponsibleAI ResourcesandtoolstointegrateResponsibleAIpracticesintoyourMLworkflow Community Join Explorewaystogetinvolved Blog StayuptodatewithallthingsTensorFlow Forum↗ DiscussionplatformfortheTensorFlowcommunity Groups Usergroups,interestgroupsandmailinglists Contribute Guideforcontributingtocodeanddocumentation WhyTensorFlow About Casestudies English 中文–简体 GitHub Signin Addons Overview Guide&Tutorials API Install Learn More API More Resources More Overview Guide&Tutorials API Community More WhyTensorFlow More GitHub tfa Overviewregister_all tfa.activations Overviewgeluhardshrinklishtmishrrelusnakesoftshrinksparsemaxtanhshrink tfa.callbacks OverviewAverageModelCheckpointTQDMProgressBarTimeStopping tfa.image Overviewadjust_hsv_in_yiqangles_to_projective_transformsblendcompose_transformsconnected_componentscutoutdense_image_warpequalizeeuclidean_dist_transformgaussian_filter2dinterpolate_bilinearinterpolate_splinemean_filter2dmedian_filter2drandom_cutoutrandom_hsv_in_yiqresamplerrotatesharpnessshear_xshear_ysparse_image_warptransformtranslatetranslate_xytranslations_to_projective_transforms tfa.layers OverviewAdaptiveAveragePooling1DAdaptiveAveragePooling2DAdaptiveAveragePooling3DAdaptiveMaxPooling1DAdaptiveMaxPooling2DAdaptiveMaxPooling3DCRFCorrelationCostESNEmbeddingBagFilterResponseNormalizationGELUGroupNormalizationInstanceNormalizationMaxUnpooling2DMaxUnpooling2DV2MaxoutMultiHeadAttentionNoisyDensePoincareNormalizePolynomialCrossingSnakeSparsemaxSpatialPyramidPooling2DSpectralNormalizationStochasticDepthTLUWeightNormalization tfa.losses OverviewContrastiveLossGIoULossLiftedStructLossNpairsLossNpairsMultilabelLossPinballLossSigmoidFocalCrossEntropySparsemaxLossTripletHardLossTripletSemiHardLossWeightedKappaLosscontrastive_lossgiou_losslifted_struct_lossnpairs_lossnpairs_multilabel_losspinball_losssigmoid_focal_crossentropysparsemax_losstriplet_hard_losstriplet_semihard_loss tfa.metrics OverviewCohenKappaF1ScoreFBetaScoreGeometricMeanHammingLossHarmonicMeanKendallsTauMatthewsCorrelationCoefficientMeanMetricWrapperMultiLabelConfusionMatrixRSquarehamming_distancehamming_loss_fn tfa.optimizers OverviewAdaBeliefAdamWAveragedOptimizerWrapperCOCOBConditionalGradientCyclicalLearningRateDecoupledWeightDecayExtensionExponentialCyclicalLearningRateLAMBLazyAdamLookaheadMovingAverageMultiOptimizerNovoGradProximalAdagradRectifiedAdamSGDWSWATriangular2CyclicalLearningRateTriangularCyclicalLearningRateYogiextend_with_decoupled_weight_decay tfa.options Overviewdisable_custom_kernelenable_custom_kernelis_custom_kernel_disabled tfa.rnn OverviewESNCellLayerNormLSTMCellLayerNormSimpleRNNCellNASCellPeepholeLSTMCell tfa.seq2seq OverviewAttentionMechanismAttentionWrapperAttentionWrapperStateBahdanauAttentionBahdanauMonotonicAttentionBaseDecoderBasicDecoderBasicDecoderOutputBeamSearchDecoderBeamSearchDecoderOutputBeamSearchDecoderStateCustomSamplerDecoderFinalBeamSearchDecoderOutputGreedyEmbeddingSamplerInferenceSamplerLuongAttentionLuongMonotonicAttentionSampleEmbeddingSamplerSamplerScheduledEmbeddingTrainingSamplerScheduledOutputTrainingSamplerSequenceLossTrainingSamplerdynamic_decodegather_treegather_tree_from_arrayhardmaxmonotonic_attentionsafe_cumprodsequence_losstile_batch tfa.text OverviewCRFModelWrapperCrfDecodeForwardRnnCellcrf_binary_scorecrf_constrained_decodecrf_decodecrf_decode_backwardcrf_decode_forwardcrf_filtered_inputscrf_forwardcrf_log_likelihoodcrf_log_normcrf_multitag_sequence_scorecrf_sequence_scorecrf_unary_scoreparse_timeskip_gram_sampleskip_gram_sample_with_text_vocabviterbi_decode crf Overview tfa.types OverviewAcceptableDTypesActivationConstraintFloatTensorLikeInitializerNumberOptimizerRegularizerTensorLike Introduction TensorFlow ForJavaScript ForMobile&Edge ForProduction TensorFlow(v2.10.0) Versions… TensorFlow.js TensorFlowLite TFX Models&datasets Tools Libraries&extensions TensorFlowCertificateprogram LearnML ResponsibleAI Join Blog Forum↗ Groups Contribute About Casestudies TensorFlow Resources Addons API Module:tfa Stayorganizedwithcollections Saveandcategorizecontentbasedonyourpreferences. ViewsourceonGitHub UsefulextrafunctionalityforTensorFlowmaintainedbySIG-addons. Modules activationsmodule:Additionalactivationfunctions. callbacksmodule:AdditionalcallbacksthatconformtoKerasAPI. imagemodule:Additionalimagemanipulationops. layersmodule:AdditionallayersthatconformtoKerasAPI. lossesmodule:AdditionallossesthatconformtoKerasAPI. metricsmodule:AdditionalmetricsthatconformtoKerasAPI. optimizersmodule:AdditionaloptimizersthatconformtoKerasAPI. optionsmodule rnnmodule:AdditionalRNNcellsthatcorformtoKerasAPI. seq2seqmodule:Additionallayersforsequencetosequencemodels. textmodule:Additionaltext-processingops. typesmodule:Typesfortypingfunctionssignatures. Functions register_all(...):RegisterTensorFlowAddons'objectsinTensorFlowglobaldictionaries. Exceptasotherwisenoted,thecontentofthispageislicensedundertheCreativeCommonsAttribution4.0License,andcodesamplesarelicensedundertheApache2.0License.Fordetails,seetheGoogleDevelopersSitePolicies.JavaisaregisteredtrademarkofOracleand/oritsaffiliates. Lastupdated2022-06-03UTC. [{ "type":"thumb-down", "id":"missingTheInformationINeed", "label":"MissingtheinformationIneed" },{ "type":"thumb-down", "id":"tooComplicatedTooManySteps", "label":"Toocomplicated/toomanysteps" },{ "type":"thumb-down", "id":"outOfDate", "label":"Outofdate" },{ "type":"thumb-down", "id":"samplesCodeIssue", "label":"Samples/codeissue" },{ "type":"thumb-down", "id":"otherDown", "label":"Other" }] [{ "type":"thumb-up", "id":"easyToUnderstand", "label":"Easytounderstand" },{ "type":"thumb-up", "id":"solvedMyProblem", "label":"Solvedmyproblem" },{ "type":"thumb-up", "id":"otherUp", "label":"Other" }] Stayconnected Blog Forum GitHub Twitter YouTube Support Issuetracker Releasenotes StackOverflow Brandguidelines CiteTensorFlow Terms Privacy SignupfortheTensorFlownewsletter Subscribe English 中文–简体