public static TensorShapeProto as_shape <T>(T[] dims)
{
TensorShapeProto shape = new TensorShapeProto();
for (int i = 0; i < dims.Length; i++)
{
var dim = new TensorShapeProto.Types.Dim();
switch (dims[i])
{
case int n:
dim.Size = n;
break;
case long l:
dim.Size = l;
break;
default:
throw new NotImplementedException("as_shape Not Implemented");
}
// dim.Name = $"dim_{i}";
shape.Dim.Add(dim);
}
return(shape);
}
public static TensorShapeProto as_proto(this Shape tshape)
{
TensorShapeProto shape = new TensorShapeProto();
for (int i = 0; i < tshape.ndim; i++)
{
var dim = new TensorShapeProto.Types.Dim();
dim.Size = tshape.dims[i];
//dim.Name = $"dim_{i}";
shape.Dim.Add(dim);
}
return(shape);
}
public static TensorShapeProto as_shape(long[] dims)
{
TensorShapeProto shape = new TensorShapeProto();
for (int i = 0; i < dims.Length; i++)
{
var dim = new TensorShapeProto.Types.Dim();
dim.Size = dims[i];
dim.Name = $"dim_{i}";
shape.Dim.Add(dim);
}
return(shape);
}
