python - 如何在渴望模式下访问张量值
问题描述
我在我的数据集上使用地图功能。在映射的函数中,我想访问张量的值以在“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)
推荐阅读
- ansible - 特定任务应该只针对单个服务器
- c# - Chromium 如何知道/设置打开的 pdf 文件中的滚动位置?
- amazon-web-services - codepipeline.putJobSuccessResult() Hanging when Invoking a Lambda from VPC
- ios - 如何在 Xcode 中显示照片库图像?
- c++ - 期望:代码打印出x(您输入的)出现在矩阵中的次数...实际:代码打印出值的数量
- java - “--continue”标志的Gradle属性?
- r - H2o Rd 警告:“缺少文件链接”
- lisp - lisp 可以在 lisp 中实现这样的糖语法吗?
- javascript - 与 FireFox 和 Chrome 相比,Edge 和 Safari 上的鼠标事件监听器行为异常,知道为什么吗?
- ruby-on-rails - Restangular 帖子错误地创建了工资计划的单数