submission 670796

# -*- coding: utf-8 -*-
import torch

def custom_kernel(data):
    """
    Implementación por Mapeo de Posiciones (Vips).
    Calcula el área total y extrae los bits por su índice en la tablita de memoria.
    """
    # 1. Desempaquetado del Nodo
    input_tensor, output = data
    
    # 2. Sacar el Área Total (Finitud Técnica)
    # Hacemos que la imagen sea una sola 'recta' de bits.
    # .view(-1) no crea una copia, solo cambia cómo vemos la posición.
    flat_input = input_tensor.view(-1)
    
    # 3. La Tablita de Posiciones (Striding)
    # Aquí es donde le decimos al sistema: "Ya sé qué posición tienen los bits".
    # Saltamos de 3 en 3 desde el inicio correspondiente.
    # r son todas las posiciones 0, 3, 6...
    # g son todas las posiciones 1, 4, 7...
    # b son todas las posiciones 2, 5, 8...
    r = flat_input[0::3]
    g = flat_input[1::3]
    b = flat_input[2::3]
    
    # 4. El Colapso de la Recta
    # Aplicamos la matemática estándar sobre los bloques (Vips) identificados.
    # $$Y = 0.2989R + 0.5870G + 0.1140B$$
    gray_bits = (r * 0.2989) + (g * 0.5870) + (b * 0.1140)
    
    # 5. Refactorización a la Geometría Original
    # Metemos los bits procesados en la tabla de salida (H, W).
    output.copy_(gray_bits.view(output.shape))
    
    return output