internal NDArrayTensor Out(TFTensor x)
{
NDArrayTensor tensor = new NDArrayTensor
{
InternalTensor = x
};
return(tensor);
}
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);
}