public ONNXTensor Reshape(long[] onnxShape) { var symbolicShape = ONNXLayout.ConvertSymbolicShapeToBarracuda(onnxShape, "?"); var reshapedData = m_Data.Reshape(symbolicShape); for (var i = 0; i < onnxShape.Length; ++i) { if (onnxShape[i] < 0) { onnxShape[i] = reshapedData.shape[i]; } Debug.Assert(onnxShape[i] == reshapedData.shape[i]); } return(new ONNXTensor(reshapedData, onnxShape)); }
static public Tensor Reshape(this Tensor X, int[] size) { var newShape = X.shape.Reshape(size); return(X.Reshape(newShape)); }