hash
paddle.fluid.layers.
hash
(input, hash_size, num_hash=1, name=None)[源代码]
该OP将输入 hash 成为一个整数,该数的值小于给定的 hash_size
。仅支持输入为LoDTensor。
该OP使用的哈希算法是:xxHash - Extremely fast hash algorithm
- 参数:
- input (Variable) - 输入是一个 二维
LoDTensor
。输入维数必须为2。数据类型为:int32、int64。仅支持LoDTensor。 - hash_size (int) - 哈希算法的空间大小。输出值将保持在 范围内。
- num_hash (int) - 哈希次数。默认值为1。
- name (str,可选) – 具体用法请参见 Name ,一般无需设置,默认值为None。
- input (Variable) - 输入是一个 二维
返回:LoDTensor
返回类型:Variable
代码示例:
- import paddle.fluid as fluid
- import numpy as np
- place = fluid.core.CPUPlace()
- # 构建网络
- x = fluid.layers.data(name="x", shape=[1], dtype="int32", lod_level=1)
- res = fluid.layers.hash(name="res",input=x, hash_size=1000, num_hash=4)
- # 创建CPU执行器
- exe = fluid.Executor(place)
- exe.run(fluid.default_startup_program())
- in1 = np.array([[1,2],[3,4]]).astype("int32")
- print(in1)
- x_i = fluid.core.LoDTensor()
- x_i.set(in1,place)
- x_i.set_recursive_sequence_lengths([[0,2]])
- res = exe.run(fluid.default_main_program(), feed={'x':x_i}, fetch_list=[res], return_numpy=False)
- print(np.array(res[0]))
- # [[[722]
- # [407]
- # [337]
- # [395]]
- # [[603]
- # [590]
- # [386]
- # [901]]]