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