测试tensoflow or pytorch代码

Tensorflow

show一个例子这里，注意打印tf变量前先执行sess.run(tf.global_variables_initializer())以初始化变量。该例子包含如何定义变量以及单元测试。

import tensorflow as tf
def weights_nonzero(labels):
    # """Assign weight 1.0 to all labels except for padding (id=0)."""
    return tf.to_float(tf.not_equal(labels, 0))

if __name__ == "__main__":
	x = tf.Variable([[1,2,3,4,0,0,0], [3,4,5,6,7,0,0]])
	y= weights_nonzero(x)
	init=tf.global_variables_initializer()
	with tf.Session() as sess:
	    sess.run(init)
	    print(sess.run([x, y]))

Pytorch

show一个例子，pytorch的测试比较直观，所见即所得。在此主要记录几个创建变量的方法，用于做测试用例。

def a_b_c(a, b, c):
	print(a)
	print(b)
	print(c)

if __name__ == "__main__":
	#初始化整型变量，0-5内取值，size=(2,4)，转类型为long
	inp = torch.randint(0, 5, (2, 4), dtype=torch.long) 
	#第二个tensor直接复制
	inp_1 = copy.deepcopy(inp)
	#创建连续数值的tensor
	inp_2 = torch.arange(0, 8).view(2, 4)
	a_b_c_(inp, inp_1, inp_2)