Uses of Class
org.tribuo.util.onnx.ONNXInitializer
Packages that use 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
Methods in org.tribuo.math.onnx that return ONNXInitializerModifier and TypeMethodDescriptionstatic ONNXInitializerONNXMathUtils.floatMatrix(ONNXContext context, String name, Matrix matrix, boolean transpose) Builds aONNXInitializercontaining theMatrix.static ONNXInitializerONNXMathUtils.floatVector(ONNXContext context, String name, SGDVector vector) Builds aONNXInitializercontaining theSGDVector. -
Uses of ONNXInitializer in org.tribuo.util.onnx
Methods in org.tribuo.util.onnx that return ONNXInitializerModifier 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).