How do I get numpy related functions to run on a GPU?

1

How do I get numpy related functions to run on a GPU

2

PyTorch implements a lot of numpy-equivalent functions, which can be executed on a GPU, if the data is already pushed to the GPU.
Are you missing some numpy methods or which functions would you like to use?

3

I want to use np.dot.

4

You could use torch.dot. You can find a lot of functions in the docs.

5

But the dot function only applies to 1 dimension.

6

As stated in the numpy docs for multi-dimensional arrays other methods (e.g. matmul) should be preferred, which are also implemented in PyTorch.