sample_nonoverlapping_windows_new_ax#

hypercoil.functional.window.sample_nonoverlapping_windows_new_ax(tensor: Tensor, window_size: int, num_windows: int = 1, windowing_axis: int = -1, multiplying_axis: int = 0, *, key: PRNGKey) Tensor[source]#

Sample windows from a tensor.

Parameters:
tensorTensor

The tensor to sample from.

window_sizeint

The size of the window(s) to sample.

num_windowsint, optional

The number of windows to sample. Default: 1.

windowing_axisint, optional

The axis along which to sample windows. Default: -1.

multiplying_axisint, optional

The axis along which to multiply the windows. Default: 0.

Returns:
Tensor

Windows from the input tensor stacked along the multiplying axis.