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)