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)