コード例 #1
0
ファイル: utils.py プロジェクト: jackd/weighpoint 
def map_gather(params, indices, **kwargs):
    return maybe_ragged_lambda_call(_map_gather, [params, indices],
                                    arguments=kwargs)
コード例 #2
0
ファイル: utils.py プロジェクト: jackd/weighpoint 
def apply_row_offset(indices, offset):
    return maybe_ragged_lambda_call(_apply_row_offset, [indices, offset])
コード例 #3
0
ファイル: utils.py プロジェクト: jackd/weighpoint 
def reduce_sum(tensor, axis=None, keepdims=False):
    return maybe_ragged_lambda_call(_utils.reduce_sum,
                                    tensor,
                                    arguments=dict(axis=axis,
                                                   keepdims=keepdims))
コード例 #4
0
ファイル: utils.py プロジェクト: jackd/weighpoint 
def get_row_offsets(tensor):
    return maybe_ragged_lambda_call(_utils.get_row_offsets, tensor)
コード例 #5
0
ファイル: utils.py プロジェクト: jackd/weighpoint 
def merge_features(node_features, edge_features):
    return maybe_ragged_lambda_call(_utils.merge_features,
                                    [node_features, edge_features])
コード例 #6
0
ファイル: utils.py プロジェクト: jackd/weighpoint 
def reshape_final_dim(tensor, final_dims):
    return maybe_ragged_lambda_call(_utils.reshape_final_dim,
                                    tensor,
                                    arguments=dict(final_dims=final_dims))
コード例 #7
0
ファイル: utils.py プロジェクト: jackd/weighpoint 
def flatten_leading_dims(tensor, n=2):
    return maybe_ragged_lambda_call(_utils.flatten_leading_dims,
                                    tensor,
                                    arguments=dict(n=n))