说明:我就废话不多说了,直接上代码吧!
import tensorflow as tf
def model_1():
with tf.variable_scope("var_a"):
a = tf.Variable(initial_value=[1, 2, 3], name="a")
vars = [var for var in tf.trainable_variables() if var.name.startswith("var_a")]
print(len(vars))
<weixin_38663443> 上传 | 大小:28kb
说明:如下所示:
#coding=gbk
'''
GPU上面的环境变化太复杂,这里我直接给出在笔记本CPU上面的运行时间结果
由于方式3需要将tensor转换到GPU上面,这一过程很消耗时间,大概需要十秒,故而果断抛弃这样的做法
img (168, 300, 3)
sub div in numpy,time 0.0110
sub div in torch.tensor,time 0.0070
sub div in torch.tensor with torchvision.transforms,t <weixin_38570145> 上传 | 大小:71kb