Function conv2d_weight_backward 

pub fn conv2d_weight_backward(
    x: Tensor<4>,
    weight: Tensor<4>,
    output_grad: Tensor<4>,
    options: ConvOptions<2>,
) -> Tensor<4>

Calculate the 2D convolution backward pass, returning the gradient for weight.