腾讯云
开发者社区
文档
建议反馈
控制台
首页
学习
活动
专区
工具
TVP
最新优惠活动
文章/答案/技术大牛
搜索
搜索
关闭
发布
登录/注册
精选内容/技术社群/优惠产品,
尽在小程序
立即前往
文章
问答
(9999+)
视频
沙龙
1
回答
JAX
中有
多个
‘
vmap
`?
python
、
vectorization
、
jax
import
jax
.numpy as jnp return jnp.sqrt为了计算xt、yt和zt中每对数据点的梯度,我必须执行以下操作: fn_prime =
vmap
(grad(fn, argnums=(0, 1, 2)), in_axes=(None, None, None
浏览 22
提问于2021-10-17
得票数 1
1
回答
矢量嵌套
vmap
python
、
numpy
、
jax
以下是我掌握的一些数据:import numpyro.distributions as dist
jax
.
vmap
(lambda axis:
jax
.
vmap
(func, (None, 0))(axis, yaxis))(xaxis)另一种方
浏览 5
提问于2021-11-04
得票数 2
回答已采纳
1
回答
在
Jax
(谷歌)
中有
没有类似的CUDA threadId?
jax
我正在尝试理解
jax
.
vmap
/pmap (
jax
:https://
jax
.readthedocs.io/)的行为。CUDA有threadId让你知道哪个线程在执行代码,在
jax
中有
类似的概念吗?(
jax
.process_id没有)
浏览 61
提问于2021-10-05
得票数 1
回答已采纳
2
回答
有效地从函数填充数组
python
、
numpy
、
jax
我想从一个函数构造一个2D数组,这样我就可以利用
jax
.jit了。 我通常使用numpy的方法是创建一个空数组,然后就地填充该数组。xx[ii_a, ii_b] = xyz[0] zz[ii_a, ii_b] = xyz[2] 为了在
jax
中实现这一点,我尝试使用
jax
.opt.index_update。yy = yy.at[ii_a, ii_b].set(xyz[1]) zz = zz.at[ii_a, ii_b].set(xyz[2]) 它运
浏览 33
提问于2020-10-06
得票数 1
回答已采纳
1
回答
如何将
jax
vmap
用于嵌套循环?
python
、
performance
、
vectorization
、
jit
、
jax
为了提高性能,我想使用
vmap
来向量化这段代码。b in dataB] for a in dataA]) return func(x, y) mapped =
jax
.
vmap
浏览 24
提问于2021-10-04
得票数 3
回答已采纳
1
回答
使用
vmap
(
jax
)对矩阵进行元素求和?
python
、
jax
我正在尝试理解
vmap
中的in_axes和out_axes选项。例如,我想要将两个矩阵相加,得到具有相同形状的输出。X = np.arange(9).reshape(3,3)def sum2(x,y):
vmap
ipython-input-403-103694166574> in <module> 4 return x + y ----> 5
vmap</
浏览 33
提问于2021-07-11
得票数 2
回答已采纳
1
回答
具有Python /Autograd的向量值函数的Jacobian行列式
python
、
numpy
、
automatic-differentiation
、
jax
输入/ (现在我将坚持使用Autograd,它提供了一个autograd.jacobian()方法,但只要得到我想要的东西,我就很乐意使用
JAX
)。**( )( >>> import autograd.numpy as np>>> x = np.array(
浏览 5
提问于2020-01-14
得票数 2
回答已采纳
2
回答
如何在压缩参数上使用
Jax
?
python
、
jax
., dtype=float32)]
jax
.
vmap
(f)(zip(xs, ys))ValueError:
vmap
was requested to map its argument along axis
浏览 9
提问于2022-06-05
得票数 1
回答已采纳
1
回答
用于简单数组更新的
Jax
python
、
jax
我是
Jax
的新手,我正在努力转换别人的代码,它使用了numba的“快速数学”特性,并且依赖于许多嵌套的for-循环,而不造成太多的性能损失。我正在尝试使用
Jax
的
vmap
函数重新创建相同的行为。下面是我试图用
vmap
向量化的一个简单的例子:from
jax
import
vmap
a = jnp.arange(20(a,
jax
.ops.index[i,
浏览 3
提问于2021-05-23
得票数 2
回答已采纳
1
回答
JAX
是否单独改变了
JAX
函数的性能?
jit
、
jax
我正在学习使用
JAX
,我对jit和
vmap
的使用有一些疑问,因为我无法通过阅读文档来解决这些问题。@
jax
.jit def fooBar(x):返回foo(x) + bar(x) 如果foo()和bar()已经被抛弃了,有什么区别吗?,,在我把它放进
vmap
之后,我应该设置一个函数吗?在上面的例子中,我应该做
jax
.
vmap
(fooBar)还是
jax<
浏览 3
提问于2021-06-25
得票数 2
1
回答
vmapped函数与vmapped函数输出之间的区别
python
、
jax
我正在阅读一套很好的
JAX
笔记本,其中包含关于的练习。我感兴趣的任务是通过
多个
vmap
函数应用程序复制以下函数: s =
vmap
(np.sum)(cp)我最初的“猜测解决方案”是尝试: func1 = <em
浏览 5
提问于2022-10-09
得票数 1
回答已采纳
1
回答
一批数据集上的
Jax
-
vmap
python
、
jax
、
flax
在
JAX
中,我希望
vmap
是一个固定长度的数据类型列表上的函数,例如:from flax import struct classparam_list = jnp.Array([EnvParams(max_steps=500), EnvParams(max_steps=600)])上面的示例失败了,因为无法创建自定义对象的jnp.Array,而且
JAX
不允许在Py
浏览 15
提问于2022-09-18
得票数 0
1
回答
从堆叠矩阵中获取批索引- Python
python
、
matrix
、
jax
5,5,2,4]], [[1,2,3,4], ]) e =
jax
.
vmap
浏览 9
提问于2022-05-23
得票数 1
回答已采纳
1
回答
Python -不同的正则/分析函数
python
、
numpy
、
scipy
numpy as np a = x**3+5 g2i =
jax
.
vmap
(
jax
.grad(
jax
.grad(f)))(xi) g3i =
jax
.
vmap
(
jax
.grad(
jax
.grad(<
浏览 3
提问于2021-07-30
得票数 0
回答已采纳
1
回答
Jax
中的
vmap
ops.index_update
python
、
performance
、
numpy
、
optimization
、
jax
我只是想知道是否有办法实现
vmap
?def filter_
jax
(y): y =
jax
.ops.index_update(y,
jax
.ops.index[(jnpData).block_until_ready()def paraUpdate(y, ind):
浏览 0
提问于2020-11-03
得票数 1
回答已采纳
1
回答
如何利用
JAX
vmap
有效地计算重要抽样估计
python
、
reinforcement-learning
、
jax
在使用
JAX
进行优化之前,此函数的代码如下: """ Calculate the unweighted importanceresult[ii] = pi_ratio_prod * weighted_return 不幸的是,我不能仅仅重写这个函数来处理一个事件,然后使用
jax
.
vmap
原因是我想要矢量化的参数是一个定制的Episode对象,
JAX
不
浏览 2
提问于2022-10-01
得票数 0
回答已采纳
1
回答
JAX
vmap
行为
python
、
vectorization
、
jax
我试图理解
JAX
vmap
的行为,所以我编写了以下代码:from
jax
import
vmap
z = jnp.dot(a,b)b = jnp.array
浏览 10
提问于2021-03-09
得票数 2
回答已采纳
2
回答
对
JAX
中多维数组上接受多维输入的函数进行矢量化
python
、
arrays
、
jax
下面是一个示例:from
jax
import map return jnp.dot(A, B) R =
vmap
(我希望结果是形状(10, 10, 2, 2),因为我认为
vmap
将函数应用于指定轴的每个子数组。显然,我没有正确地理解它是如何
浏览 4
提问于2022-04-10
得票数 1
1
回答
JAX
vmap
中的in_axes关键字
jax
、
auto-vectorization
我正在尝试使用
vmap
来理解
JAX
的自动矢量化功能,并根据
JAX
的文档实现了一个最小的工作示例。import
jax
.numpy as jnp input_batch = jnp.ones((batc
浏览 8
提问于2022-01-03
得票数 4
回答已采纳
1
回答
在
JAX
中,elementwise_grad的选择是什么?
python
、
neural-network
、
automatic-differentiation
、
jax
对于自动区分,我使用
JAX
库。为了计算我的目标变量'u‘的一阶和二阶导数,即计算du/dx和d2u/dx2,在一个例子中使用了elementwise_grad。在
jax
中,它的替代方案是什么?通过
JAX
,我应该使用哪些函数来计算du/dx和d2u/dx2?
浏览 4
提问于2021-12-08
得票数 1
点击加载更多
扫码
添加站长 进交流群
领取专属
10元无门槛券
手把手带您无忧上云
相关
资讯
关于要替代TensorFlow的JAX,你知道多少?
Excel 中有100多个隐藏的工作表,如何批量取消隐藏?
吸星大法取长补短,机器学习界要“取代”numpy的JAX库
Excel 中有多个条件格式时,某些靠左某些靠右才能不重叠, but how?
非洲50多个国家中有十余个适合电子商务发展的环境
热门
标签
更多标签
云服务器
即时通信 IM
ICP备案
对象存储
实时音视频
活动推荐
运营活动
广告
关闭
领券