3 from __future__
import absolute_import
4 from __future__
import division
5 from __future__
import print_function
6 from __future__
import unicode_literals
13 def rand_array(*dims):
16 return np.array(np.random.rand(*dims) - 0.5).astype(np.float32)
22 workspace.GlobalInit([
24 '--caffe2_log_level=0',
28 core.SetEnginePref({}, {})
31 self.
ws = workspace.C.Workspace()
32 workspace.ResetWorkspace()
35 workspace.ResetWorkspace()