Contents

squeeze(axis:)

Adds a squeeze operation in the graph.

Declaration

func squeeze(axis: Int) -> BNNSGraph.Builder.Tensor<T>

Parameters

  • axis:

    The axis at which the operation removes the dimension

Discussion

This operation deletes a dimension of size 1 at the given index.