import torch


def generate_const_input(model, device):
    """Return constant inputs shape = (n_different_inputs, input_size)"""
    return torch.eye(model.input_size, dtype=torch.float32, device=device)