pub struct ConfidenceOutput {
pub plddt_logits: Tensor,
pub plddt: Tensor,
pub pae_logits: Option<Tensor>,
pub pde_logits: Option<Tensor>,
pub distogram_logits: Option<Tensor>,
}Expand description
Output of the confidence head.
Fields§
§plddt_logits: TensorpLDDT logits [B, N_tok, num_plddt_bins].
plddt: TensorpLDDT scalar per token [B, N_tok], range [0, 1].
pae_logits: Option<Tensor>pAE logits [B, N_tok, N_tok, num_pae_bins]. None if not produced.
pde_logits: Option<Tensor>pDE logits [B, N_tok, N_tok, num_pde_bins]. None if not produced.
distogram_logits: Option<Tensor>Distogram logits [B, N_tok, N_tok, distogram_bins]. None if not produced.
Auto Trait Implementations§
impl Freeze for ConfidenceOutput
impl !RefUnwindSafe for ConfidenceOutput
impl Send for ConfidenceOutput
impl Sync for ConfidenceOutput
impl Unpin for ConfidenceOutput
impl UnsafeUnpin for ConfidenceOutput
impl !UnwindSafe for ConfidenceOutput
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more
§impl<T> Identity for Twhere
T: ?Sized,
impl<T> Identity for Twhere
T: ?Sized,
§impl<T> Instrument for T
impl<T> Instrument for T
§fn instrument(self, span: Span) -> Instrumented<Self>
fn instrument(self, span: Span) -> Instrumented<Self>
§fn in_current_span(self) -> Instrumented<Self>
fn in_current_span(self) -> Instrumented<Self>
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self>
fn into_either(self, into_left: bool) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read more