From ea741affa11d0cbc9efe1719cfe19c1568cbaca1 Mon Sep 17 00:00:00 2001 From: zmxdream Date: Mon, 26 Dec 2022 23:37:15 +0800 Subject: [PATCH] add float16 to index_sample eng doc (#49317) --- python/paddle/tensor/search.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/paddle/tensor/search.py b/python/paddle/tensor/search.py index 76d1ec705ee..5dfe526e5ef 100755 --- a/python/paddle/tensor/search.py +++ b/python/paddle/tensor/search.py @@ -734,7 +734,7 @@ def index_sample(x, index): Args: x (Tensor): The source input tensor with 2-D shape. Supported data type is - int32, int64, float32, float64. + int32, int64, float16, float32, float64. index (Tensor): The index input tensor with 2-D shape, first dimension should be same with X. Data type is int32 or int64. -- GitLab