tfa.losses.contrastive_loss | TensorFlow Addons

Computes the contrastive loss between y_true and y_pred . ... This loss encourages the embedding to be close to each other for the samples of the ... 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 tfa.losses.contrastive_loss Stayorganizedwithcollections Saveandcategorizecontentbasedonyourpreferences. ViewsourceonGitHub Computesthecontrastivelossbetweeny_trueandy_pred. @tf.function tfa.losses.contrastive_loss( y_true:tfa.types.TensorLike, y_pred:tfa.types.TensorLike, margin:tfa.types.Number=1.0 )->tf.Tensor Thislossencouragestheembeddingtobeclosetoeachotherfor thesamplesofthesamelabelandtheembeddingtobefarapartatleast bythemarginconstantforthesamplesofdifferentlabels. Theeuclideandistancesy_predbetweentwoembeddingmatrices aandbwithshape[batch_size,hidden_size]canbecomputed asfollows: a=tf.constant([[1,2], [3,4], [5,6]],dtype=tf.float16) b=tf.constant([[5,9], [3,6], [1,8]],dtype=tf.float16) y_pred=tf.linalg.norm(a-b,axis=1) y_pred See:http://yann.lecun.com/exdb/publis/pdf/hadsell-chopra-lecun-06.pdf Args y_true 1-DintegerTensorwithshape[batch_size]of binarylabelsindicatingpositivevsnegativepair. y_pred 1-DfloatTensorwithshape[batch_size]of distancesbetweentwoembeddingmatrices. margin marginterminthelossdefinition. Returns contrastive_loss 1-DfloatTensorwithshape[batch_size]. 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 中文–简体