fThX$SrSSKrSSKJr SSKJr SSKJrJ r J r J r J r SSK rSSKrSSKrSSKJr SSKJrJrJr SS KJr SS KJrJr SS KJrJr SS KJrJ r SS K!J"r"J#r#J$r$ SSK%J&r&J'r' SSK(J)r) \$RT"\+5r,\"SS\"55r-\"SS\"55r.Sr/"SS\R`5r1"SS\R`5r2S>S\R`S\RfS\RfS\RfS\ \RfS\4S \44S!jjr5"S"S#\R`5r6"S$S%\R`5r7"S&S'\R`5r8"S(S)\R`5r9"S*S+\R`5r:"S,S-\R`5r;"S.S/\R`5r<\#"S0S1\55r=\#"S2S3\=55r>"S4S5\R`5r?\#"S6S79"S8S9\=55r@\#"S:S79"S;S<\=55rA/S=QrBg)?z,PyTorch VideoMAE (masked autoencoder) model.N)deepcopy) dataclass)CallableOptionalSetTupleUnion)nn)BCEWithLogitsLossCrossEntropyLossMSELoss)ACT2FN)BaseModelOutputImageClassifierOutput)ALL_ATTENTION_FUNCTIONSPreTrainedModel) find_pruneable_heads_and_indicesprune_linear_layer) ModelOutputauto_docstringlogging)IMAGENET_DEFAULT_MEANIMAGENET_DEFAULT_STD)VideoMAEConfigc\rSrSr%SrSr\\R\ S'Sr \\ \R\ S'Sr \\ \R\ S'Sr g)VideoMAEDecoderOutput,a Class for VideoMAEDecoder's outputs, with potential hidden states and attentions. Args: logits (`torch.FloatTensor` of shape `(batch_size, patch_size ** 2 * num_channels)`): Pixel reconstruction logits. hidden_states (`tuple(torch.FloatTensor)`, *optional*, returned when `output_hidden_states=True` is passed or when `config.output_hidden_states=True`): Tuple of `torch.FloatTensor` (one for the output of the embeddings + one for the output of each layer) of shape `(batch_size, sequence_length, hidden_size)`. Hidden-states of the model at the output of each layer plus the initial embedding outputs. attentions (`tuple(torch.FloatTensor)`, *optional*, returned when `output_attentions=True` is passed or when `config.output_attentions=True`): Tuple of `torch.FloatTensor` (one for each layer) of shape `(batch_size, num_heads, sequence_length, sequence_length)`. Attentions weights after the attention softmax, used to compute the weighted average in the self-attention heads. Nlogits hidden_states attentions)__name__ __module__ __qualname____firstlineno____doc__r rtorch FloatTensor__annotations__r!rr"__static_attributes__r#f/var/www/auris/envauris/lib/python3.13/site-packages/transformers/models/videomae/modeling_videomae.pyrr,sR +/FHU&& '.8