xlnstorch.operators.lns_max_pool3d

xlnstorch.operators.lns_max_pool3d(x, kernel_size, base, stride=None, padding=0, dilation=1, ceil_mode=False, return_indices=False)

See docs for torch.nn.functional.max_pool3d() for parameter/return details. Typically, torch.Tensor arguments are the equivalent internal representations of LNStensors.

Parameters:

• x (Tensor)

• kernel_size (int | Tuple[int])

• base (Tensor)

• stride (int | Tuple[int] | None)

• padding (int | Tuple[int])

• dilation (int | Tuple[int])

• ceil_mode (bool)

• return_indices (bool)

Return type:

Tensor