phygnn.layers.custom_layers.WindowGeometry

class WindowGeometry(batch_size: Tensor, query_height: int | Tensor, query_width: int | Tensor, window_size: int | Tensor, window_shift: int | Tensor, query_height_padding: int | Tensor, query_width_padding: int | Tensor, padded_query_height: int | Tensor, padded_query_width: int | Tensor, n_window_rows: int | Tensor, n_window_cols: int | Tensor, n_windows: int | Tensor, time_steps: int | Tensor = 1)[source]

Bases: object

Computed layout for one windowed-attention call.

Methods

Attributes

`time_steps`
`batch_size`
`query_height`
`query_width`
`window_size`
`window_shift`
`query_height_padding`
`query_width_padding`
`padded_query_height`
`padded_query_width`
`n_window_rows`
`n_window_cols`
`n_windows`