Optional gradient direction in `SobelGradients`

`SobelGradients` currently calculate the gradients in both horizontal and vertical directions. However, there are use cases that only one of these gradients in required, so it will have redundant computations. It would be nice to be able to set `direction` and calculate the gradient only in the direction.