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); }