


from ...configuration_utils import PretrainedConfig
from ...utils import logging


UpperCamelCase__ =						logging.get_logger(__name__)

UpperCamelCase__ =						{
    "facebook/timesformer": "https://huggingface.co/facebook/timesformer/resolve/main/config.json",
}
class 			_lowerCAmelCase   ( _UpperCAmelCase			):






			"""simple docstring"""





			lowercase__      :  int      =			"""timesformer"""





			def __init__(				self	:       Optional[int]			,     lowercase	:       Dict=224			,     lowercase	:       Optional[int]=16			,     lowercase	:       Tuple=3			,     lowercase	:       List[str]=8			,     lowercase	:       List[str]=768			,     lowercase	:       Any=12			,     lowercase	:       Optional[Any]=12			,     lowercase	:       Tuple=3_072			,     lowercase	:       Any="gelu"			,     lowercase	:       Any=0.0			,     lowercase	:       Tuple=0.0			,     lowercase	:       Any=0.02			,     lowercase	:       Optional[Any]=1E-6			,     lowercase	:       Union[str, Any]=True			,     lowercase	:       Tuple="divided_space_time"			,     lowercase	:       List[Any]=0			,     **lowercase	:       str			,     )						->					List[str]:
									"""simple docstring"""







									super().__init__(**lowercase   )

									__lowercase					=				image_size
									__lowercase					=				patch_size
									__lowercase					=				num_channels
									__lowercase					=				num_frames

									__lowercase					=				hidden_size
									__lowercase					=				num_hidden_layers
									__lowercase					=				num_attention_heads
									__lowercase					=				intermediate_size
									__lowercase					=				hidden_act
									__lowercase					=				hidden_dropout_prob
									__lowercase					=				attention_probs_dropout_prob
									__lowercase					=				initializer_range
									__lowercase					=				layer_norm_eps
									__lowercase					=				qkv_bias

									__lowercase					=				attention_type
									__lowercase					=				drop_path_rate










import unittest

import numpy as np






def  UpperCAmelCase__    (    lowercase__						,       lowercase__						,       lowercase__						,       lowercase__ = None						,       ) ->      np.ndarray:
						__lowercase					=				np.shape(lowercase__   )
						__lowercase					=				np.shape(lowercase__   )
						__lowercase					=				np.shape(lowercase__   )

						if shape_a[0] != shape_b[0]:
												__lowercase					=				(
												    """Expected the same number of rows for A and B. """
												    F"Instead found A of size {shape_a} and B of size {shape_b}"
												)
												raise ValueError(lowercase__   )

						if shape_b[1] != shape_c[1]:
												__lowercase					=				(
												    """Expected the same number of columns for B and C. """
												    F"Instead found B of size {shape_b} and C of size {shape_c}"
												)
												raise ValueError(lowercase__   )

						__lowercase					=				pseudo_inv
						if a_inv is None:
												try:
																		__lowercase					=				np.linalg.inv(lowercase__   )
												except np.linalg.LinAlgError:
																		raise ValueError(
																		    """Input matrix A is not invertible. Cannot compute Schur complement."""   )

						return mat_c - mat_b.T @ a_inv @ mat_b
class 			_lowerCAmelCase   ( unittest.TestCase			):






			"""simple docstring"""





			def 		snake_case__			(				self	:       Dict   )						->					None:
									"""simple docstring"""







									__lowercase					=				np.array([[1, 2, 1], [2, 1, 2], [3, 2, 4]]   )
									__lowercase					=				np.array([[0, 3], [3, 0], [2, 3]]   )
									__lowercase					=				np.array([[2, 1], [6, 3]]   )

									__lowercase					=				schur_complement(lowercase			,     lowercase			,     lowercase   )

									__lowercase					=				np.block([[a, b], [b.T, c]]   )

									__lowercase					=				np.linalg.det(lowercase   )
									__lowercase					=				np.linalg.det(lowercase   )
									__lowercase					=				np.linalg.det(lowercase   )

									self.assertAlmostEqual(lowercase			,     det_a * det_s   )





			def 		snake_case__			(				self	:       Tuple   )						->					None:
									"""simple docstring"""







									__lowercase					=				np.array([[1, 2, 1], [2, 1, 2], [3, 2, 4]]   )
									__lowercase					=				np.array([[0, 3], [3, 0], [2, 3]]   )
									__lowercase					=				np.array([[2, 1], [6, 3]]   )

									with self.assertRaises(lowercase   ):
															schur_complement(lowercase			,     lowercase			,     lowercase   )







			def 		snake_case__			(				self	:       Tuple   )						->					None:
									"""simple docstring"""







									__lowercase					=				np.array([[1, 2, 1], [2, 1, 2], [3, 2, 4]]   )
									__lowercase					=				np.array([[0, 3], [3, 0], [2, 3]]   )
									__lowercase					=				np.array([[2, 1, 3], [6, 3, 5]]   )

									with self.assertRaises(lowercase   ):
															schur_complement(lowercase			,     lowercase			,     lowercase   )


if __name__ == "__main__":
			import doctest

			doctest.testmod()
			unittest.main()





