gpt4 book ai didi

python - 从维度为 [N] 的张量创建维度为 [N,D] 的矩阵

转载 作者:太空宇宙 更新时间:2023-11-03 15:46:21 24 4
gpt4 key购买 nike

我有一个维度为 N 的张量,我想复制它以创建一个维度为 NxD 的张量,每一列都是初始向量。

谢谢

最佳答案

您首先要将张量扩展/ reshape 为 N x 1 形状,然后在二维中将其平铺 D 次:

tensor_N_x_1 = tf.expand_dims(tensor, 1)     # Expand by adding a dim in position 1
tensor_N_x_D = tf.tile(tensor_N_x_1, [1, D]) # Tile 1 time in the 1st dim, D times in the 2nd

文档:

关于python - 从维度为 [N] 的张量创建维度为 [N,D] 的矩阵,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/49709616/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com