a label and a feature tensor. The shape of the label tensor

Input:

Output:

Prompt: