示例#1
0
        internal NDArrayTensor Out(TFTensor x)
        {
            NDArrayTensor tensor = new NDArrayTensor
            {
                InternalTensor = x
            };

            return(tensor);
        }
示例#2
0
        internal NDArrayTensor Out(TFOutput x, bool castFloat = false)
        {
            NDArrayTensor tensor = new NDArrayTensor();

            if (castFloat)
            {
                tensor.InternalTensor = InternalEval(tf.Cast(x, TFDataType.Float));
            }
            else
            {
                tensor.InternalTensor = InternalEval(x);
            }

            return(tensor);
        }