Class ModelMetadataExtensions
Model metadata extensions
Namespace: Unity.Barracuda
Syntax
public static class ModelMetadataExtensions : object
Methods
GetDownStreamLayersCount(Model, String)
Get count of layers that directly depend on specified input
Declaration
public static int GetDownStreamLayersCount(this Model model, string name)
Parameters
Type | Name | Description |
---|---|---|
Model | model | Model |
String | name | input name |
Returns
Type | Description |
---|---|
Int32 | count of layers that directly depend on specified input |
GetShapeByName(Model, String)
Get model tensor shape by name
Declaration
public static TensorShape? GetShapeByName(this Model model, string name)
Parameters
Type | Name | Description |
---|---|---|
Model | model | Model |
String | name | Tensor name |
Returns
Type | Description |
---|---|
Nullable<TensorShape> | Tensor shape |
GetTensorByName(Model, String)
Get model tensor by name
Declaration
public static Tensor GetTensorByName(this Model model, string name)
Parameters
Type | Name | Description |
---|---|---|
Model | model | Model |
String | name | Tensor name |
Returns
Type | Description |
---|---|
Tensor | Tensor |