submission 66424

from task import input_t, output_t
import torch
import torch.nn.functional as F
from utils import DeterministicContext


def custom_kernel(data: input_t) -> output_t:
    """
    Implementation of 2D convolution using PyTorch with no padding and no striding.
    Args:
        data: Tuple of (input tensor, kernel tensor)
        spec: Convolution specifications
    Returns:
        Output tensor after convolution
    """
    with DeterministicContext():
        input_tensor, kernel, output = data
        output[...] = F.conv2d(input_tensor, kernel, stride=1, padding=0)
        return output