tensorflow2.0 numpy.ndarray 与tenor直接互转

论坛 期权论坛 编程之家     
选择匿名的用户   2021-5-29 23:40   23   0

1、代码参考

import numpy as np
import tensorflow as tf

a = np.random.random((5,3)) b = np.random.randint(0,9,(3,1)) c = tf.tensordot(a.astype(np.float),b.astype(np.float),axes=1) # tensor 转ndarray dn = c.numpy() print(dn) # ndarray转tensor tn = tf.convert_to_tensor(dn) print(tn)

转载于:https://www.cnblogs.com/wuzaipei/p/10956279.html

分享到 :
0 人收藏
您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

积分:3875789
帖子:775174
精华:0
期权论坛 期权论坛
发布
内容

下载期权论坛手机APP