Uses of Class
org.tribuo.util.onnx.ONNXInitializer
Package
Description
Utilities for converting Tribuo math objects into ONNX representations.
Interfaces and utilities for writing ONNX models from Java.
-
Uses of ONNXInitializer in org.tribuo.math.onnx
Modifier and TypeMethodDescriptionstatic ONNXInitializer
ONNXMathUtils.floatMatrix
(ONNXContext context, String name, Matrix matrix, boolean transpose) Builds aONNXInitializer
containing theMatrix
.static ONNXInitializer
ONNXMathUtils.floatVector
(ONNXContext context, String name, SGDVector vector) Builds aONNXInitializer
containing theSGDVector
. -
Uses of ONNXInitializer in org.tribuo.util.onnx
Modifier and TypeMethodDescriptionCreates a float tensor for this ONNXContext, populated according to parameters.Creates a tensor for this ONNXContext, populated according to parameters.Creates a float tensor for this ONNXContext, populated according to parameters.Creates an int tensor for this ONNXContext, populated according to parameters.Creates a long tensor for this ONNXContext, populated according to parameters.Creates a float scalar constant for this ONNXContext.Creates a long scalar constant for this ONNXContext.ONNXContext.floatTensor
(String baseName, List<Integer> dims, Consumer<FloatBuffer> populate) Creates a tensor for this ONNXContext, populated asONNXUtils.floatTensorBuilder(ONNXContext, String, List, Consumer)
.