func.func @transpose_2d() {
  %input = util.unfoldable_constant dense<[[1, 2, 3],
                                           [4, 5, 6]]> : tensor<2x3xi32>
  %0 = "stablehlo.transpose"(%input) {
    permutation = array<i64: 1, 0>
  } : (tensor<2x3xi32>) -> tensor<3x2xi32>
  check.expect_eq_const(%0, dense<[[1, 4],
                                   [2, 5],
                                   [3, 6]]> : tensor<3x2xi32>) : tensor<3x2xi32>
  return
}

func.func @transpose_3d() {
  %input = util.unfoldable_constant dense<[[[ 1,  2,  3],
                                            [ 4,  5,  6]],
                                           [[ 7,  8,  9],
                                            [10, 11, 12]]]> : tensor<2x2x3xi32>
  %0 = "stablehlo.transpose"(%input) {
    permutation = array<i64: 0, 2, 1>
  } : (tensor<2x2x3xi32>) -> tensor<2x3x2xi32>
  check.expect_eq_const(%0, dense<[
    [[ 1,  4],
     [ 2,  5],
     [ 3,  6]],
    [[ 7, 10],
     [ 8, 11],
     [ 9, 12]]]> : tensor<2x3x2xi32>) : tensor<2x3x2xi32>
  return
}