torch.floor_divide¶
- torch.floor_divide(input, other, *, out=None) Tensor ¶
注意
在 PyTorch 1.13 之前,
torch.floor_divide()
错误地执行了截断除法。要恢复以前的 behavior,请使用torch.div()
并设置rounding_mode='trunc'
。按元素计算
input
除以other
,并对结果进行向下取整。支持广播到公共形状、类型提升以及整数和浮点数输入。
示例
>>> 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])