Sample Nodes by specific types Args: count: A scalar tensor specify sample count types: A scalar tensor specify sample type condition: A atrribute specify sample condition Return: A 1-d tensor of sample node ids
(count, node_type, condition='')
| 36 | |
| 37 | |
| 38 | def sample_node(count, node_type, condition=''): |
| 39 | """ |
| 40 | Sample Nodes by specific types |
| 41 | |
| 42 | Args: |
| 43 | count: A scalar tensor specify sample count |
| 44 | types: A scalar tensor specify sample type |
| 45 | condition: A atrribute specify sample condition |
| 46 | |
| 47 | Return: |
| 48 | A 1-d tensor of sample node ids |
| 49 | """ |
| 50 | if node_type == '-1': |
| 51 | types = -1 |
| 52 | else: |
| 53 | types = type_ops.get_node_type_id(node_type) |
| 54 | return base._LIB_OP.sample_node(count, types, condition) |
| 55 | |
| 56 | |
| 57 | def sample_edge(count, edge_type=None): |