From 977960a41204cb6a86bbd5bbb9420425576141d1 Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Sun, 24 Nov 2019 15:16:58 +0100
Subject: [PATCH] Add get_random_sampling
---
src/pystencils_autodiff/transformations.py | 9 +++++++++
1 file changed, 9 insertions(+)
diff --git a/src/pystencils_autodiff/transformations.py b/src/pystencils_autodiff/transformations.py
index 17d1c20..ada9fa4 100644
--- a/src/pystencils_autodiff/transformations.py
+++ b/src/pystencils_autodiff/transformations.py
@@ -1,5 +1,7 @@
import itertools
+
import sympy as sp
+from sympy.matrices.dense import matrix_multiply_elementwise
import pystencils
from pystencils import Field, x_vector
@@ -32,3 +34,10 @@ def add_fixed_constant_boundary_handling(assignments, with_cse=True):
return safe_assignments
else:
return pystencils.AssignmentCollection(safe_assignments)
+
+
+def get_random_sampling(random_numbers, aabb_min, aabb_max):
+ random_numbers = sp.Matrix(random_numbers)
+ aabb_min = sp.Matrix(aabb_min)
+ aabb_max = sp.Matrix(aabb_max)
+ return matrix_multiply_elementwise(random_numbers, (aabb_max - aabb_min)) + aabb_min
--
GitLab