def mnp_swapaxes(input_tensor):
a = mnp.swapaxes(input_tensor, 0, 1)
b = mnp.swapaxes(input_tensor, 1, 0)
c = mnp.swapaxes(input_tensor, 1, 1)
d = mnp.swapaxes(input_tensor, 2, 1)
e = mnp.swapaxes(input_tensor, 1, 2)
f = mnp.swapaxes(input_tensor, 2, 2)
return a, b, c, d, e, f
def test_swapaxes_exception():
with pytest.raises(ValueError):
mnp.swapaxes(mnp.ones((3, 3)), 1, 10)
def construct(self, x):
x = mnp.rollaxis(x, 2)
x = mnp.swapaxes(x, 0, 1)
return x
def swapaxes_exception(input_tensor):
return mnp.swapaxes(input_tensor, 1, 10)
def swapaxes(data, axis0, axis1):
return mnp.swapaxes(data, axis0, axis1)
def t(data):
validator.check_equal_int(data.ndim, 2)
return mnp.swapaxes(data, 1, 0)
