提交 af6f3c04 编写于 作者: Q qiaolongfei

use float32 in cond_op

上级 0900aedf
......@@ -15,7 +15,7 @@ class PySimpleCond(object):
for i in range(1, 10, 2):
array[i] = 0
self.cond = np.array(array)
self.x = np.ones(shape=(10, 1))
self.x = np.ones(shape=(10, 1)).astype("float32")
def forward(self):
self.index_t = np.where(self.cond == 1)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册