I am working on wrapping a function written in C++ to PyTorch. The C++ function takes 3 floats and produces 1 float.
Is there a C++ PyTorch API to perform the broadcasting of the wrapped function?
Probably something like np.nditer in numpy.