shanyi15-patch-3
Kexin Zhao 7 years ago
parent 5a10d833db
commit 8cce330430

@ -134,7 +134,7 @@ def infer(use_cuda, inference_program, save_dirname=None):
lod = [[3, 4, 2]]
base_shape = [1]
# The range of random integers is [low, high]
tensor_words = fluid.create_random_lodtensor(
tensor_words = fluid.create_random_int_lodtensor(
lod, base_shape, place, low=0, high=len(word_dict) - 1)
results = inferencer.infer({'words': tensor_words})
print("infer results: ", results)

@ -149,7 +149,7 @@ def infer(use_cuda, inference_program, save_dirname=None):
lod = [[3, 4, 2]]
base_shape = [1]
# The range of random integers is [low, high]
tensor_words = fluid.create_random_lodtensor(
tensor_words = fluid.create_random_int_lodtensor(
lod, base_shape, place, low=0, high=len(word_dict) - 1)
results = inferencer.infer({'words': tensor_words})
print("infer results: ", results)

@ -141,7 +141,7 @@ def infer(use_cuda, inference_program, save_dirname=None):
lod = [[3, 4, 2]]
base_shape = [1]
# The range of random integers is [low, high]
tensor_words = fluid.create_random_lodtensor(
tensor_words = fluid.create_random_int_lodtensor(
lod, base_shape, place, low=0, high=len(word_dict) - 1)
results = inferencer.infer({'words': tensor_words})
print("infer results: ", results)

@ -247,7 +247,7 @@ def infer(word_dict, use_cuda, save_dirname=None):
lod = [[3, 4, 2]]
base_shape = [1]
# The range of random integers is [low, high]
tensor_words = fluid.create_random_lodtensor(
tensor_words = fluid.create_random_int_lodtensor(
lod, base_shape, place, low=0, high=word_dict_len - 1)
# Construct feed as a dictionary of {feed_target_name: feed_target_data}

Loading…
Cancel
Save