Package | Description |
---|---|
org.bytedeco.tensorrt.nvinfer |
Modifier and Type | Method and Description |
---|---|
IScatterLayer |
INetworkDefinition.addScatter(ITensor data,
ITensor indices,
ITensor updates,
int mode) |
IScatterLayer |
VNetworkDefinition.addScatter(ITensor data,
ITensor indices,
ITensor updates,
int mode) |
IScatterLayer |
INetworkDefinition.addScatter(ITensor data,
ITensor indices,
ITensor updates,
nvinfer.ScatterMode mode)
\brief Add a Scatter layer to the network with specified mode and axis=0.
|
IScatterLayer |
VNetworkDefinition.addScatter(ITensor data,
ITensor indices,
ITensor updates,
nvinfer.ScatterMode mode) |
Copyright © 2024. All rights reserved.