torch.distributions.LogitsToProbsOptions
export interface LogitsToProbsOptions {
/** If true, uses sigmoid. If false, uses softmax. Default: true */
is_binary?: boolean;
}is_binary(boolean)optional- – If true, uses sigmoid. If false, uses softmax. Default: true
Options for logits_to_probs