reshape(shape:source:)
Adds a new reshape layer to the graph using the shape and source tensor you specify.
Declaration
func reshape(shape: [Int], source: MLCTensor) -> MLCTensor?Parameters
- shape:
An array representing the shape of result tensor.
- source:
The source tensor.
Return Value
A result tensor.