Package | Description |
---|---|
org.bytedeco.pytorch | |
org.bytedeco.pytorch.global |
Modifier and Type | Method and Description |
---|---|
LossReduction |
LossReduction.put(kMean value) |
LossReduction |
LossReduction.put(kNone value) |
LossReduction |
LossReduction.put(kSum value) |
LossReduction |
LossReduction.put(LossReduction x) |
LossReduction |
HuberLossOptions.reduction() |
LossReduction |
BCEWithLogitsLossOptions.reduction() |
LossReduction |
MultiMarginLossOptions.reduction() |
LossReduction |
MSELossOptions.reduction() |
LossReduction |
SoftMarginLossOptions.reduction() |
LossReduction |
L1LossOptions.reduction() |
LossReduction |
CTCLossOptions.reduction() |
LossReduction |
MultiLabelSoftMarginLossOptions.reduction() |
LossReduction |
TripletMarginWithDistanceLossOptions.reduction() |
LossReduction |
MarginRankingLossOptions.reduction() |
LossReduction |
MultiLabelMarginLossOptions.reduction() |
LossReduction |
CosineEmbeddingLossOptions.reduction() |
LossReduction |
CrossEntropyLossOptions.reduction() |
LossReduction |
BCELossOptions.reduction() |
LossReduction |
SmoothL1LossOptions.reduction() |
LossReduction |
HingeEmbeddingLossOptions.reduction() |
LossReduction |
TripletMarginLossOptions.reduction() |
LossReduction |
PoissonNLLLossOptions.reduction() |
LossReduction |
NLLLossOptions.reduction() |
Modifier and Type | Method and Description |
---|---|
static kNone |
LossReduction.get0(LossReduction container) |
static kMean |
LossReduction.get1(LossReduction container) |
static kSum |
LossReduction.get2(LossReduction container) |
LossReduction |
LossReduction.put(LossReduction x) |
Modifier and Type | Method and Description |
---|---|
static Tensor |
torch.binary_cross_entropy_with_logits(Tensor input,
Tensor target,
Tensor weight,
LossReduction reduction,
Tensor pos_weight) |
static Tensor |
torch.binary_cross_entropy(Tensor input,
Tensor target,
Tensor weight,
LossReduction reduction) |
static Tensor |
torch.cosine_embedding_loss(Tensor input1,
Tensor input2,
Tensor target,
double margin,
LossReduction reduction) |
static Tensor |
torch.cross_entropy(Tensor input,
Tensor target,
Tensor weight,
long ignore_index,
LossReduction reduction,
double label_smoothing) |
static Tensor |
torch.ctc_loss(Tensor log_probs,
Tensor targets,
Tensor input_lengths,
Tensor target_lengths,
long blank,
LossReduction reduction,
boolean zero_infinity) |
static Tensor |
torch.hinge_embedding_loss(Tensor input,
Tensor target,
double margin,
LossReduction reduction) |
static Tensor |
torch.huber_loss(Tensor input,
Tensor target,
LossReduction reduction) |
static Tensor |
torch.huber_loss(Tensor input,
Tensor target,
LossReduction reduction,
double delta) |
static Tensor |
torch.l1_loss(Tensor input,
Tensor target,
LossReduction reduction)
Options for
torch::nn::functional::binary_cross_entropy_with_logits . |
static Tensor |
torch.margin_ranking_loss(Tensor input1,
Tensor input2,
Tensor target,
double margin,
LossReduction reduction) |
static Tensor |
torch.mse_loss(Tensor input,
Tensor target,
LossReduction reduction) |
static Tensor |
torch.multi_margin_loss(Tensor input,
Tensor target,
long p,
double margin,
Tensor weight,
LossReduction reduction) |
static Tensor |
torch.multilabel_margin_loss(Tensor input,
Tensor target,
LossReduction reduction) |
static Tensor |
torch.multilabel_soft_margin_loss(Tensor input,
Tensor target,
Tensor weight,
LossReduction reduction) |
static Tensor |
torch.nll_loss(Tensor input,
Tensor target,
Tensor weight,
long ignore_index,
LossReduction reduction) |
static Tensor |
torch.poisson_nll_loss(Tensor input,
Tensor target,
boolean log_input,
boolean full,
double eps,
LossReduction reduction) |
static Tensor |
torch.smooth_l1_loss(Tensor input,
Tensor target,
LossReduction reduction) |
static Tensor |
torch.smooth_l1_loss(Tensor input,
Tensor target,
LossReduction reduction,
double beta) |
static Tensor |
torch.soft_margin_loss(Tensor input,
Tensor target,
LossReduction reduction) |
static Tensor |
torch.triplet_margin_loss(Tensor anchor,
Tensor positive,
Tensor negative,
double margin,
double p,
double eps,
boolean swap,
LossReduction reduction) |
static Tensor |
torch.triplet_margin_with_distance_loss(Tensor anchor,
Tensor positive,
Tensor negative,
Pointer distance_function,
double margin,
boolean swap,
LossReduction reduction) |
Copyright © 2024. All rights reserved.