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. |