xlnstorch.operators.lns_avg_pool2d

xlnstorch.operators.lns_avg_pool2d(x, kernel_size, stride=None, padding=0, ceil_mode=False, count_include_pad=True, divisor_override=None)

See docs for torch._C._nn.avg_pool2d() for parameter/return details. Typically, torch.Tensor arguments are the equivalent internal representations of LNStensors.

Parameters:

• x (Tensor)

• kernel_size (int)

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

• padding (int | Tuple[int])

• ceil_mode (bool)

• count_include_pad (bool)

• divisor_override (Tensor | None)

Return type:

Tensor