TensorFlow多维矩阵的转置,tensorflow.transpose()的理解

TensorFlow——多维矩阵的转置(transpose):
https://blog.csdn.net/qq_37174526/article/details/80905693

这一篇写得相当清楚。

我额外添加一些对参数perm的理解:


这是官方文档里的描述,简单的翻译:

转置矩阵a根据参数perm对矩阵维数进行调整。

返回的张量的维数i将对应于输入维数perm[i]。如果没有指定perm,默认为(n-1…0),其中n是输入张量的秩。因此,默认情况下,这个操作对二维输入张量执行一个常规矩阵转置。如果共轭(conjugate)为true,a.dtype可以是complex64,也可以是complex128,然后对a的值进行共轭和转置。

在numpy转置中,@compatibility(numpy)是一种内存效率高的常量时间操作,因为它们只是用调整后的步长返回相同数据的新视图。张量流不支持大步,因此转置返回一个新的张量,其中的项被置换。@end_compatibility

perm[i]内放的是转置后的新张量的轴相对于旧张量的改变,也就是你想要的改变。

初始情况下,n维张量中轴的排列为[0,1,2,3,…,n-1]
在这里插入图片描述
例如:一个二维张量x,初始情况下轴的排列为(0,1)

[
	[1, 2, 3], 
	[4, 5, 6]
	]

perm=[1, 0],意味着新矩阵中 [轴0变为轴1,轴1变为轴0].

 x = tf.constant([[1, 2, 3], [4, 5, 6]]) tf.transpose(x) 
#[[1, 4]#[2, 5] # [3, 6]]

 Equivalently(相当于):

 tf.transpose(x, perm=[1, 0]) 
 #[[1, 4] 
 # [2, 5] # [3, 6]]

我在实际中遇到的例子:

>>>X=tf.constant(housing_data_plus_bias,dtype=tf.float32,name="X")
>>>y=tf.constant(housing.target.reshape(-1,1),dtype=tf.float32,name="y")
>>>print(X.shape)
(20640, 9)

>>>XT=tf.transpose(X)#张量转置
>>>print(XT)
Tensor("transpose:0", shape=(9, 20640), dtype=float32)
  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值