torch.floor_divide

torch.floor_divide(input, other, *, out=None) → Tensor

Примечание

Перед PyTorch 1.13 torch.floor_divide() неправильно выполняло деление с усечением. Чтобы восстановить предыдущее поведение, используйте torch.div() с rounding_mode='trunc'.

Вычисляет input делённое на other, поэлементно, и округляет результат в меньшую сторону.

$$\text{{out}}_i = \text{floor} \left( \frac{{\text{{input}}_i}}{{\text{{other}}_i}} \right)$$

Поддерживает вещание к общему размеру, продвижение типов, а также целочисленные и плавающие входные данные.

Параметры:

• input (Тензор или Число) – делимое
• other (Тензор или Число) – делитель

Ключевые аргументы:

out (Тензор, необязательно) – выходной тензор.

Пример:

>>> a = torch.tensor([4.0, 3.0])
>>> b = torch.tensor([2.0, 2.0])
>>> torch.floor_divide(a, b)
tensor([2.0, 1.0])
>>> torch.floor_divide(a, 1.4)
tensor([2.0, 2.0])