tensorflow如何实现数组过滤?

Question

如何使用tensorflow实现在numpy中的功能，numpy示例如下:a=np.array[-1,2,3,-2]a[a>0]=1a[a<0]=0...

Answer 1

import tensorflow as tf# 创建一个常量op， 产生一个1x2矩阵，这个op被作为一个节点# 加到默认视图中# 构造器的返回值代表该常量op的返回值matrix1 = tr.constant([[3., 3.]])# 创建另一个常量op, 产生一个2x1的矩阵matrix2 = tr.constant([[2.], [2.]])# 创建一个矩阵乘法matmul op，把matrix1和matrix2作为输入：product = tf.matmul(matrix1, matrix2)