Is there a tvm function that supports the same operation as pytorch nn.unsqueeze?

#1

Is there a tvm function that supports the same operation as pytorch nn.unsqueeze?

It’s confirmed that there’s a squeeze. but i can’t find unsqueeze operation?

Do you have any calculations to get the same result as this?

#2

is it expand_dims what you want?