CollectionDef should cover most collections.
To add a user-defined collection, do one of the following:
1. For simple data types, such as string, int, float:
tf.add_to_collection("your_collection_name", your_simple_value)
strings will be stored as bytes_list.
2. For Protobuf types, there are three ways to add them:
1) tf.add_to_collection("your_collection_name",
your_proto.SerializeToString())
collection_def {
key: "user_defined_bytes_collection"
value {
bytes_list {
value: "queue_name: \"test_queue\"\n"
}
}
}
or
2) tf.add_to_collection("your_collection_name", str(your_proto))
collection_def {
key: "user_defined_string_collection"
value {
bytes_list {
value: "\n\ntest_queue"
}
}
}
or
3) any_buf = any_pb2.Any()
tf.add_to_collection("your_collection_name",
any_buf.Pack(your_proto))
collection_def {
key: "user_defined_any_collection"
value {
any_list {
value {
type_url: "type.googleapis.com/tensorflow.QueueRunnerDef"
value: "\n\ntest_queue"
}
}
}
}
3. For Python objects, implement to_proto() and from_proto(), and register
them in the following manner:
ops.register_proto_function("your_collection_name",
proto_type,
to_proto=YourPythonObject.to_proto,
from_proto=YourPythonObject.from_proto)
These functions will be invoked to serialize and de-serialize the
collection. For example,
ops.register_proto_function(ops.GraphKeys.GLOBAL_VARIABLES,
proto_type=variable_pb2.VariableDef,
to_proto=Variable.to_proto,
from_proto=Variable.from_proto)
tensorflow.CollectionDef de type Protobuf.CollectionDef Classes imbriquées
| classe | CollectionDef.AnyList | AnyList is used for collecting Any protos. | |
| interface | CollectionDef.AnyListOrBuilder | ||
| classe | CollectionDef.Builder | CollectionDef should cover most collections. | |
| classe | CollectionDef.BytesList | BytesList is used for collecting strings and serialized protobufs. | |
| interface | CollectionDef.BytesListOrBuilder | ||
| classe | CollectionDef.FloatList | FloatList is used for collecting float values. | |
| interface | CollectionDef.FloatListOrBuilder | ||
| classe | CollectionDef.Int64List | Int64List is used for collecting int, int64 and long values. | |
| interface | CollectionDef.Int64ListOrBuilder | ||
| énumération | CollectionDef.KindCase | ||
| classe | CollectionDef.NodeList | NodeList is used for collecting nodes in graph. | |
| interface | CollectionDef.NodeListOrBuilder | ||
Constantes
| int | ANY_LIST_FIELD_NUMBER | |
| int | BYTES_LIST_FIELD_NUMBER | |
| int | FLOAT_LIST_FIELD_NUMBER | |
| int | INT64_LIST_FIELD_NUMBER | |
| int | NODE_LIST_FIELD_NUMBER |
Méthodes publiques
| booléen | est égal (Obj objet) |
| CollectionDef.AnyList | getAnyList () .tensorflow.CollectionDef.AnyList any_list = 5; |
| CollectionDef.AnyListOrBuilder | getAnyListOrBuilder () .tensorflow.CollectionDef.AnyList any_list = 5; |
| CollectionDef.BytesList | getBytesList () .tensorflow.CollectionDef.BytesList bytes_list = 2; |
| CollectionDef.BytesListOrBuilder | getBytesListOrBuilder () .tensorflow.CollectionDef.BytesList bytes_list = 2; |
| CollectionDef statique | |
| CollectionDef | |
| final statique com.google.protobuf.Descriptors.Descriptor | |
| CollectionDef.FloatList | getFloatList () .tensorflow.CollectionDef.FloatList float_list = 4; |
| CollectionDef.FloatListOrBuilder | getFloatListOrBuilder () .tensorflow.CollectionDef.FloatList float_list = 4; |
| CollectionDef.Int64List | getInt64Liste () .tensorflow.CollectionDef.Int64List int64_list = 3; |
| CollectionDef.Int64ListOrBuilder | getInt64ListOrBuilder () .tensorflow.CollectionDef.Int64List int64_list = 3; |
| CollectionDef.KindCase | getKindCase () |
| CollectionDef.NodeList | getNodeList () .tensorflow.CollectionDef.NodeList node_list = 1; |
| CollectionDef.NodeListOrBuilder | getNodeListOrBuilder () .tensorflow.CollectionDef.NodeList node_list = 1; |
| int | |
| final com.google.protobuf.UnknownFieldSet | |
| booléen | hasAnyList () .tensorflow.CollectionDef.AnyList any_list = 5; |
| booléen | hasBytesList () .tensorflow.CollectionDef.BytesList bytes_list = 2; |
| booléen | hasFloatList () .tensorflow.CollectionDef.FloatList float_list = 4; |
| booléen | hasInt64Liste () .tensorflow.CollectionDef.Int64List int64_list = 3; |
| booléen | hasNodeList () .tensorflow.CollectionDef.NodeList node_list = 1; |
| int | |
| booléen final | |
| CollectionDef.Builder statique | newBuilder (prototype CollectionDef ) |
| CollectionDef.Builder statique | |
| CollectionDef.Builder | |
| CollectionDef statique | parseDelimitedFrom (entrée InputStream) |
| CollectionDef statique | parseDelimitedFrom (entrée InputStream, com.google.protobuf.ExtensionRegistryLite extensionRegistry) |
| CollectionDef statique | parseFrom (données ByteBuffer) |
| CollectionDef statique | parseFrom (entrée com.google.protobuf.CodedInputStream, com.google.protobuf.ExtensionRegistryLite extensionRegistry) |
| CollectionDef statique | parseFrom (données ByteBuffer, com.google.protobuf.ExtensionRegistryLite extensionRegistry) |
| CollectionDef statique | parseFrom (entrée com.google.protobuf.CodedInputStream) |
| CollectionDef statique | parseFrom (données octet[], com.google.protobuf.ExtensionRegistryLite extensionRegistry) |
| CollectionDef statique | parseFrom (données com.google.protobuf.ByteString) |
| CollectionDef statique | parseFrom (entrée InputStream, com.google.protobuf.ExtensionRegistryLite extensionRegistry) |
| CollectionDef statique | parseFrom (données com.google.protobuf.ByteString, com.google.protobuf.ExtensionRegistryLite extensionRegistry) |
| statique | analyseur () |
| CollectionDef.Builder | toBuilder () |
| vide | writeTo (sortie com.google.protobuf.CodedOutputStream) |
Méthodes héritées
Constantes
public statique final int ANY_LIST_FIELD_NUMBER
public statique final int BYTES_LIST_FIELD_NUMBER
public statique final int FLOAT_LIST_FIELD_NUMBER
public statique final int INT64_LIST_FIELD_NUMBER
public statique final int NODE_LIST_FIELD_NUMBER
Méthodes publiques
public booléen égal (Objet obj)
public CollectionDef.AnyListOrBuilder getAnyListOrBuilder ()
.tensorflow.CollectionDef.AnyList any_list = 5;
public CollectionDef.BytesListOrBuilder getBytesListOrBuilder ()
.tensorflow.CollectionDef.BytesList bytes_list = 2;
public statique final com.google.protobuf.Descriptors.Descriptor getDescriptor ()
public CollectionDef.FloatListOrBuilder getFloatListOrBuilder ()
.tensorflow.CollectionDef.FloatList float_list = 4;
public CollectionDef.Int64ListOrBuilder getInt64ListOrBuilder ()
.tensorflow.CollectionDef.Int64List int64_list = 3;
public CollectionDef.NodeListOrBuilder getNodeListOrBuilder ()
.tensorflow.CollectionDef.NodeList node_list = 1;
publique getParserForType ()
public int getSerializedSize ()
public final com.google.protobuf.UnknownFieldSet getUnknownFields ()
public booléen hasAnyList ()
.tensorflow.CollectionDef.AnyList any_list = 5;
public booléen hasBytesList ()
.tensorflow.CollectionDef.BytesList bytes_list = 2;
public booléen hasFloatList ()
.tensorflow.CollectionDef.FloatList float_list = 4;
public booléen hasInt64List ()
.tensorflow.CollectionDef.Int64List int64_list = 3;
public booléen hasNodeList ()
.tensorflow.CollectionDef.NodeList node_list = 1;
code de hachage int public ()
public final booléen isInitialized ()
public static CollectionDef parseDelimitedFrom (entrée InputStream, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
Jetés
| IOException |
|---|
public static CollectionDef parseFrom (entrée com.google.protobuf.CodedInputStream, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
Jetés
| IOException |
|---|
public static CollectionDef parseFrom (données ByteBuffer, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
Jetés
| InvalidProtocolBufferException |
|---|
public static CollectionDef parseFrom (entrée com.google.protobuf.CodedInputStream)
Jetés
| IOException |
|---|
public static CollectionDef parseFrom (données octet[], com.google.protobuf.ExtensionRegistryLite extensionRegistry)
Jetés
| InvalidProtocolBufferException |
|---|
public statique CollectionDef parseFrom (données com.google.protobuf.ByteString)
Jetés
| InvalidProtocolBufferException |
|---|
public static CollectionDef parseFrom (entrée InputStream, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
Jetés
| IOException |
|---|
public static CollectionDef parseFrom (données com.google.protobuf.ByteString, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
Jetés
| InvalidProtocolBufferException |
|---|
public statique analyseur ()
public void writeTo (sortie com.google.protobuf.CodedOutputStream)
Jetés
| IOException |
|---|