import svetoviz_webgpu as sv

# 1. Prepare 3D Volume (Batch=1, Channels=32, D, H, W)
# Dropout3d drops entire 3D volumetric feature maps.
dropout = nn.Dropout3d(p=0.5)

def terminal_callback(buffer, message, images, files):
    dropout.train()
    output = dropout(input_tensor)

    # Calculate volumes that were entirely zeroed
    active_vol = (output.sum(dim=(2, 3, 4)) > 0).sum().item()

    buffer.send_system_message(f"Input Volumes: {input_tensor.shape[1]}")
    buffer.send_system_message(f"Active Volumes: {active_vol} / 32")
    buffer.send_system_message("Logic: Drops independent 3D feature volumes in spatiotemporal data.")

# 2. Start the interactive session
sv.pytorch_web(module=dropout, terminal_callback=terminal_callback)