在张量流中使用tf.cond进行广播

时间:2019-01-09 22:58:57

标签: python tensorflow

假设我有一个形状=(m,n)的输入张量A。如果A [i,j]的元素大于10,则输出张量B [i,j] = 3 * A [i,j];如果元素A [i,j]小于或等于10,则B [i,j] = 2 * A [i,j]。返回B张量。在TensorFlow中实现此功能的有效方法是什么?

示例:输入A = [[1.,2。],[11。,8.]],返回B = [[2.,4。],[33。,16。]]

0 个答案:

没有答案