brainpy.math.unsqueeze

brainpy.math.unsqueeze(input, dim)

Returns a new tensor with a dimension of size one inserted at the specified position. The returned tensor shares the same underlying data with this tensor. A dim value within the range [-input.dim() - 1, input.dim() + 1) can be used. Negative dim will correspond to unsqueeze() applied at dim = dim + input.dim() + 1. :type input: Union[Array, Array] :param input: The input Array :type input: Array :type dim: int :param dim: The index at which to insert the singleton dimension :type dim: int

Returns:

out

Return type:

Array