@ProviderType
public class TensorDescriptorBuilder
extends java.lang.Object
TensorDescriptor
Modifier and Type | Method and Description |
---|---|
TensorDescriptorBuilder |
addParameter(java.lang.String name,
java.lang.Object parameter)
Add a parameter to the tensor descriptor
|
TensorDescriptor |
build()
Create an instance of TensorDescriptor
|
TensorDescriptorBuilder |
format(java.lang.String format)
Set the format of the data contained in the tensor.
|
static TensorDescriptorBuilder |
fromTensorDescriptor(TensorDescriptor descriptor)
Instantiates a builder for a
TensorDescriptor and initialises it from the supplied argument. |
TensorDescriptorBuilder |
removeParameter(java.lang.String name)
Remove a parameter from the tensor descriptor
|
public static TensorDescriptorBuilder fromTensorDescriptor(TensorDescriptor descriptor)
TensorDescriptor
and initialises it from the supplied argument.descriptor
- TensorDescriptorBuilder
.public TensorDescriptorBuilder format(java.lang.String format)
format
- a string representing the format of the data in the tensorpublic TensorDescriptorBuilder addParameter(java.lang.String name, java.lang.Object parameter)
name
- the name of the parameterparameter
- an Object representing the value of the parameterpublic TensorDescriptorBuilder removeParameter(java.lang.String name)
name
- the name of the parameterpublic TensorDescriptor build()