C3W1 Exercise 4| what to select as dtype for weight matrix

Dear all,

When initializing the weight matrix with regards to funciton trax.fastmath.random.normal(…)
I ignored the dtype and function selected the default dtype itself.

However, the unittest failed:

Output object has the incorrect type.
	Expected: <class 'jaxlib.xla_extension.DeviceArrayBase'>.
	Got: <class 'numpy.ndarray'>.
Output object has the incorrect type.
	Expected: <class 'jaxlib.xla_extension.DeviceArrayBase'>.
	Got: <class 'numpy.ndarray'>.
 8  Tests passed
 2  Tests failed

From instructions, it is said to set the default dtype for the weight matrix.
Looking for your guidance

Hi Abdirayimov,

You are not multiplying x and the weight matrix correctly.

Best,
Mubsi