首頁(yè) 猿問(wèn) tensorflow如何實(shí)現(xiàn)數(shù)組過(guò)濾?

tensorflow如何實(shí)現(xiàn)數(shù)組過(guò)濾?

人工智能

慕尼黑的夜晚無(wú)繁華 2018-10-14 15:13:28

如何使用tensorflow實(shí)現(xiàn)在numpy中的功能，numpy示例如下:a=np.array[-1,2,3,-2]a[a>0]=1a[a<0]=0...

查看完整描述

1 回答

阿晨1998

TA貢獻(xiàn)2037條經(jīng)驗(yàn) 獲得超6個(gè)贊

import tensorflow as tf
# 創(chuàng)建一個(gè)常量op，產(chǎn)生一個(gè)1x2矩陣，這個(gè)op被作為一個(gè)節(jié)點(diǎn)
# 加到默認(rèn)視圖中
# 構(gòu)造器的返回值代表該常量op的返回值
matrix1 = tr.constant([[3., 3.]])

# 創(chuàng)建另一個(gè)常量op, 產(chǎn)生一個(gè)2x1的矩陣
matrix2 = tr.constant([[2.], [2.]])

# 創(chuàng)建一個(gè)矩陣乘法matmul op，把matrix1和matrix2作為輸入：
product = tf.matmul(matrix1, matrix2)