reshape(to:)
Adds a dynamic reshape operation to the current graph.
Declaration
func reshape(to shape: some BNNSGraph.Builder.OperationParameter<Int32>) -> BNNSGraph.Builder.Tensor<T>Parameters
- shape:
A tensor that specifies the new shape.
Discussion
This function reshapes a tensor based on the values in shape.