public WeightTensor CreateWeightTensor(long[] sizes, int deviceId, bool cleanWeights = false, string name = "", IComputeGraph graphToBind = null, NormType normType = NormType.None)
{
WeightTensor r = new WeightTensor(sizes, deviceId, name, NormType: normType, graphToBind: graphToBind);
if (cleanWeights)
{
r.CleanWeight();
}
weights.Add(r);
return(r);
}
public WeightTensor CreateWeightTensor(long[] sizes, int deviceId, bool cleanWeights = false, string name = "")
{
WeightTensor r = new WeightTensor(sizes, deviceId, name);
if (cleanWeights)
{
r.CleanWeight();
}
weights.Add(r);
return(r);
}
public WeightTensor CreateWeightTensor(int row, int column, int deviceId, bool cleanWeights = false, string name = "", bool isTrainable = false, IComputeGraph graphToBind = null, NormType normType = NormType.None)
{
WeightTensor r = new WeightTensor(new long[2] {
row, column
}, deviceId, name: name, isTrainable: isTrainable, NormType: normType, graphToBind: graphToBind);
if (cleanWeights)
{
r.CleanWeight();
}
weights.Add(r);
return(r);
}
public WeightTensor CreateWeightTensor(int row, int column, int deviceId, bool cleanWeights = false, string name = "", bool isTrainable = false)
{
WeightTensor r = new WeightTensor(new long[2] {
row, column
}, deviceId, name: name, isTrainable: isTrainable);
if (cleanWeights)
{
r.CleanWeight();
}
weights.Add(r);
return(r);
}
