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)