def test_expand_dims_exception():
with pytest.raises(TypeError):
mnp.expand_dims(mnp.ones((3, 3)), 1.2)
def construct(self, x):
x = mnp.expand_dims(x, 2)
x = mnp.reshape(x, (1, 2, 3, 4, 1, 1))
x = mnp.squeeze(x)
return x
def expand_dims_exception(input_tensor):
return mnp.expand_dims(input_tensor, 1.2)
def mnp_expand_dims(input_tensor):
a = mnp.expand_dims(input_tensor, 0)
b = mnp.expand_dims(input_tensor, -1)
c = mnp.expand_dims(input_tensor, axis=2)
d = mnp.expand_dims(input_tensor, axis=-2)
return a, b, c, d
def expand_dims(data, axis):
return mnp.expand_dims(data, axis)
