Мне нужно создать трехмерный тензор, подобный этому (5,3,2), например array ([[[0, 0], [0, 1], [0, 0]], [[1, 0], [0 , 0], [0, 0]], [[0, 0], [1, 0], [.......
16 Фев 2021 в 05:59
Я хочу построить двумерный массив из функции таким образом, чтобы можно было использовать jax.jit. Обычно я делаю это с помощью numpy, создавая пустой массив, а затем заполняя этот массив на месте. xx = jnp.empty((num_a, num_b)) yy = jnp.empty((num_a, num_b)) zz = jnp.empty((num_a, num_b)) for ii_a....
6 Окт 2020 в 11:07
У меня есть два числовых массива формы (N, M). Я хотел бы вычислить строчное произведение точек. То есть создайте массив формы (N,) так, чтобы n-я строка была точечным произведением n-й строки из каждого массива. Я знаю о методе numpy inner1d. Каков наилучший способ сделать это с Jax? у jax jax.nump....
20 Апр 2020 в 05:54
Вот простой пример, который численно интегрирует произведение двух гауссовых pdf. Один из гауссианов фиксирован, среднее значение всегда равно 0. Другой гауссиан имеет среднее значение: import time import jax.numpy as np from jax import jit from jax.scipy.stats.norm import pdf # set up evaluation ....
27 Ноя 2019 в 14:00