testing
平凡搬砖
这个作者很懒,什么都没留下…
展开
-
stack与concat差异
以pytorch为例例子import torcha = torch.randint(0, 10, (2, 3, 4))b = torch.randint(0, 10, (2, 3, 4))print(a.shape, a)print(b.shape, b)c = torch.stack([a, b], dim=0)d = torch.stack([a, b], dim=1)e = torch.stack([a, b], dim=2)print(c.shape, c)print(d.s原创 2022-01-19 16:53:11 · 866 阅读 · 0 评论 -
测试tensoflow or pytorch代码
Tensorflowshow一个例子这里,注意打印tf变量前先执行sess.run(tf.global_variables_initializer())以初始化变量。该例子包含如何定义变量以及单元测试。import tensorflow as tfdef weights_nonzero(labels): # """Assign weight 1.0 to all labels except for padding (id=0).""" return tf.to_float(tf.not原创 2022-01-06 15:28:40 · 1158 阅读 · 0 评论