v0.0.4 fn (%x: Tensor[(2, 4), float32]) { let %x1: Tensor[(2, 4), float32] = nn.softmax(%x); %x1 }