| Package | Description |
|---|---|
| org.bytedeco.tensorrt.nvinfer |
| Modifier and Type | Method and Description |
|---|---|
IOneHotLayer |
INetworkDefinition.addOneHot(ITensor indices,
ITensor values,
ITensor depth,
int axis)
\brief Add a OneHot layer to the network.
|
IOneHotLayer |
VNetworkDefinition.addOneHot(ITensor indices,
ITensor values,
ITensor depth,
int axis) |
IOneHotLayer |
IOneHotLayer.getPointer(long i) |
IOneHotLayer |
IOneHotLayer.position(long position) |
Copyright © 2024. All rights reserved.