init(width:height:featureChannelCount:batchSize:)
Creates a tensor descriptor with the width and height, number of feature channels, and batch size you specify.
Declaration
convenience init?(width: Int, height: Int, featureChannelCount featureChannels: Int, batchSize: Int)Parameters
- width:
The tensor width.
- height:
The tensor height.
- featureChannels:
The number of feature channels.
- batchSize:
The tensor batch size.
Discussion
This initializer provides a convenient way for you to configure NCHW tensors used by convolutional layers. The tensor data type is MLCDataType.float32.
See Also
Creating Tensor Descriptors
init(shape:dataType:)init(shape:sequenceLengths:sortedSequences:dataType:)init(width:height:featureChannelCount:batchSize:dataType:)init(convolutionWeightsWithInputFeatureChannelCount:outputFeatureChannelCount:dataType:)init(convolutionWeightsWithWidth:height:inputFeatureChannelCount:outputFeatureChannelCount:dataType:)init(convolutionBiasesWithFeatureChannelCount:dataType:)maxTensorDimensions