[英]Access output of intermediate layers in Tensor-flow 2.0 in eager mode
[英]How to access Tensor values in eager mode
我在我的数据集上使用 map 函数。 在映射的函数中,我想访问 Tensor 的值以在“if”中使用它。
但我现在看到了访问张量的方法。
我处于急切模式并且拥有 tensorflow 2.1(因为 anaconda 不支持任何更新版本)。
这是我的意思的简单示例代码:
def f1(C):
print("every numba")
#Access C somehow
#if C < 2:
# C = C-1
return C+2
dataset = tf.data.Dataset.range(1, 6) # ==> [ 1, 2, 3, 4, 5 ]
dataset2 = dataset.map(f1)
我想这样的方法可能对你有用。
def f1(C):
print("print ", C)
if C < 2:
C = C-1
return C
dataset = tf.data.Dataset.range(1, 6) # ==> [ 1, 2, 3, 4, 5 ]
dataset = dataset.map( lambda x: tf.py_function(
f1,
inp=[x], Tout=tf.int64))
for x in dataset:
print(x)
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.