TensorFlow之张量的形状相关的操作函数主要讲解一下四种。
1、tf.shape
查看张量的形状,即返回一个一阶张量表示输入张量的形状。
tf.shape(input, name=None, out_type=tf.int32)
input:输入的张量
name:操作的名称
out_type:输出的类型(int32 or int64), 默认tf.int32
2、tf.size
返回输入张量的size,即输入张量的元素数量。
tf.size(input, name=None, out_type=tf.int32)
input:输入的张量
name:操作的名称
out_type:输出的类型 (int32 or int64),默认int32
3、tf.rank
查看张量的阶数
tf.rank(input, name=None)
input:输入的张量
name:操作的名称
4、tf.reshape
改变一个张量的形状,按照指定的shape返回一个新的张量。要是形状的某个分量是特殊值-1,那么就会计算该维度的大小,使得总大小(元素数量)保持不变,特殊的,如果传入的形状是[-1],那么意味着把整个张量弄平为1维。必须确保变形前和变形之后的总元素个数是相同的。
tf.reshape(tensor, shape, name=None)
tensor:输入的张量,待被改变形状的张量
shape:张量,必须是 int32, int64,决定了输出张量的形状
name:操作的名称
热门文章
- vue项目打包后文件过大怎么办,如何优化加载速度
- TensorFlow之张量的形状相关操作
- 1月15日19.5M/S|V2ray/SSR/Clash/Shadowrocket免费节点每天更新订阅链接,翻墙机场推荐分享
- 宠物救助站领养宠物猫(宠物救助站领养宠物要钱吗)
- OutOfMemoryError异常_在线工具
- 1月22日18.7M/S|Clash/SSR/V2ray/Shadowrocket免费节点每天更新订阅链接,翻墙机场推荐分享
- 2月8日19.3M/S|SSR/Clash/Shadowrocket/V2ray免费节点每天更新订阅链接,翻墙机场推荐分享
- java自带的四种线程池
- vue-cli3.0 吃过的坑
- 开个宠物食品店需要什么证 开个宠物食品店需要什么证件