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));
        }
예제 #2
0
        static public Tensor Reshape(this Tensor X, int[] size)
        {
            var newShape = X.shape.Reshape(size);

            return(X.Reshape(newShape));
        }