np.vectorize()

最新推荐文章于 2022-07-08 17:28:07 发布

J.Sanchez

最新推荐文章于 2022-07-08 17:28:07 发布

阅读量1.8k

点赞数

分类专栏： python 文章标签： python numpy

本文链接：https://blog.csdn.net/u010590593/article/details/124614853

版权

python 专栏收录该内容

14 篇文章

订阅专栏

函数向量化，可以对不同的元素使用不同的方法

def func(a,b):
	'''针对a,b的操作'''
	pass
inputs = np.randn((128,10))

用法

np.vectorize(func)(inputs)

例子

import torch

def func(a,b):
    if a > b:
        return a + b
    else:
        return a - b


x = torch.rand((1,3))
y = torch.rand((1,3))
print(x)
print(y)
print(np.vectorize(func)(x,y))

outputs:
tensor([[0.0838, 0.5305, 0.7352]])
tensor([[0.4186, 0.2242, 0.9994]])
[[-0.33478993  0.75468814 -0.26422805]]