I have been trying to find a pytorch equivalent for the ‘floordiv’ operator in pytorch so that I can write code that works on both NumPy Arrays and Pytorch Tensors.
Eaxmple:
a = b//c # (works with numpy, not pytorch)
a = (b/c).floor() # (works with pytorch, not numpy)
Is something equivalent present? If not, could you point me towards how to implement ‘floordiv’ in pytorch? It should be relatively easy beacause its just division followed by a floor.