public static TypeProto.Types.Tensor Initialize(this TypeProto.Types.Tensor Tensor, TensorElementType d, params int[] dims)
{
Tensor.ElemType = (int)d;
Tensor.Shape = new TensorShapeProto();
foreach (var dim in dims)
{
Tensor.Shape.Dim.Add(new TensorShapeProto.Types.Dimension()
{
DimValue = dim
});
}
return(Tensor);
}
public static string Print(this TypeProto.Types.Tensor tensor)
{
string dims = "";
foreach (var p in tensor.Shape.Dim)
{
if (dims != "")
{
dims += ",";
}
dims += p.DimValue;
}
return($"{((TensorElementType)tensor.ElemType).ToString()}[{dims}]");
}
/// <summary>Get element data type of <paramref name="tensor"/> as enum.</summary> public static TensorProto.Types.DataType ElemType(this TypeProto.Types.Tensor tensor) => (TensorProto.Types.DataType)tensor.ElemType;
public static TypeProto Initialize(this TypeProto Type, string name, TypeProto.Types.Tensor t)
{
Type.Denotation = name;
Type.TensorType = t;
return(Type);
}
