Method InferOutputDataTypes
InferOutputDataTypes(DataType[])
Infer the data types of the output tensors of the custom layer given the data types of the input tensors.
Declaration
public abstract DataType[] InferOutputDataTypes(DataType[] inputDataTypes)Parameters
| Type | Name | Description |
|---|---|---|
| DataType[] | inputDataTypes | Array of input data types. |
Returns
| Type | Description |
|---|---|
| DataType[] | Array of output data types. |