如何在Tensorflow张量中选择每行和每列的前Q个元素？
在Tensorflow中，我们可以使用tf.math.top_k函数来选择张量中每行和每列的前Q个元素。
对于选择每行的前Q个元素，我们可以使用tf.math.top_k函数的axis参数指定为1。例如，要选择张量x中每行的前Q个元素，可以使用以下代码：
import tensorflow as tf
x = tf.constant([[1, 4, 2, 5, 3],
[6, 9, 7, 8, 10]])
Q = 3 # 前3个元素
top_k_values, top_k_indices = tf.math.top_k(x, Q, sorted=True, axis=1)
print(top_k_values)
print(top_k_indices)输出结果为：
tf.Tensor(
[[ 5 4 3]
[10 9 8]], shape=(2, 3), dtype=int32)
tf.Tensor(
[[3 1 4]
[4 1 3]], shape=(2, 3), dtype=int32)对于选择每列的前Q个元素，我们可以使用tf.math.top_k函数的axis参数指定为0。例如，要选择张量x中每列的前Q个元素，可以使用以下代码：
import tensorflow as tf
x = tf.constant([[1, 4, 2, 5, 3],
[6, 9, 7, 8, 10]])
Q = 3 # 前3个元素
top_k_values, top_k_indices = tf.math.top_k(x, Q, sorted=True, axis=0)
print(top_k_values)
print(top_k_indices)输出结果为：
tf.Tensor(
[[ 6 9 7 8 10]
[ 4 4 3 5 6]
[ 2 2 2 2 3]], shape=(3, 5), dtype=int32)
tf.Tensor(
[[1 1 1 1 1]
[0 0 0 0 0]
[0 0 0 0 0]], shape=(3, 5), dtype=int32)这样，我们就可以使用Tensorflow的tf.math.top_k函数选择张量中每行和每列的前Q个元素了。
对于Tensorflow相关的产品和产品介绍链接地址，您可以参考腾讯云的文档和官方网站。
相关·内容
我在Tensorflow中实现了一个特殊的损失函数。下面是一个特殊函数的numpy风格的代码，该函数选择前Q个元素，并屏蔽每行每列中的其他元素。注意，A是一个n*n矩阵，而q是一个小于n的整数。def t
浏览 19提问于2019-06-21得票数 1
回答已采纳
2回答
我想从二维张量的每一行中提取存储在另一维张量中的列。import torchselect_tensor = tensor([1,2]) 因此，在这个特定的示例中，我想将第一行位置1中的元素(so -2)和第二行位置2中的元
浏览 9提问于2021-01-20得票数 0
回答已采纳
我有两个张量X和Z，形状为(100,)，我想创建一个张量X x Z，它将产生一个(100, 100)的形状。对于这个矩阵中的每一对元素，我想要应用一些我已经定义的函数，即fn(x，z)，用于矩阵中的每个可能的组合。我是TensorFlow新手，习惯于按顺序思考，在处理numpy数组时，每个向量上有两个<
浏览 0提问于2020-03-29得票数 3
2回答
我有一个矩阵A，它定义为tensorflow中的一个张量，由n行和p列组成。此外，我说k矩阵B1，.，Bk有p行和q列。我的目标是得到一个由n行和q列组成的矩阵C，其中C的每一行都是A中对应行的矩阵乘积，其中有一个
浏览 6提问于2016-08-11得票数 2
回答已采纳
我试着用GridLSTM和MLP/CNN训练图像。2-D image -> GridLSTM -> MLP or CNN -> GridLSTM -> MLP or CNN -> Output
我试着运行GridLSTM的示例代码和Cifar10上的教程(用于CNN)
浏览 2提问于2016-09-22得票数 1
回答已采纳
我想在keras中创建一个自定义的丢失函数。让我们调用每个示例n(即yTrue和yPred中的每一行) yT和yP。，即:对于每个样本，我总是想要计算第一个元素的平方误差，但只有当真正的标签== 1.的第一个</em
浏览 0提问于2018-05-17得票数 1
回答已采纳
如何在tensorflow中获取矩阵中每一行的最大值和次最大值？例如，我有一个矩阵张量[1,2,3,6,5,4,7,9,8]。然后我想得到每一行的最大值向量，如3,6,9和次要值向量2，4，8。问题可以分为两部分:1.找到值索引2.从矩阵中收集值并形成一个新的向量。
浏览 2提问于2018-07-16得票数 0
1回答
我正试图为人工智能中的一个问题编写一个深入的Q学习网络。我有一个函数predict()，它产生形状(None, 3)的张量，接受形状(None, 5)的输入。(None, 3)中的3对应于在每个状态下可以采取的每个动作的Q值。现在，在训练步骤中，我必须多次调用predict()并使用结果来计算成本和</
浏览 2提问于2017-08-17得票数 0
回答已采纳
我有一个有四个输出标签的tensorflow程序。我训练了这个模型，现在用它来评估不同的数据。问题是在我使用代码之后import Processdef error(forward_propagatio
浏览 1提问于2016-07-28得票数 0
1回答
正如标题所述，我试图从tensorflow中的矩阵中提取每一行的最高n个元素，并将结果存储在稀疏张量中。我已经能够用tf.nn.top_n提取索引和值，但是索引并不遵循tf.SparseTensor所要求的约定。具体来说，tf.nn.top_n返回一个与结果值矩阵(行x)形状相同的col索引矩阵，而tf.SparseTensor则需要一个(#非零
浏览 1提问于2017-04-05得票数 0
回答已采纳
我有两个张量，即X of shape (?, 32, 500)和indices of shape (?,)。对于这两个张量，0维是批处理维。indices的每个元素指定要选择的第一个维度上的X的索引。最后，我想得到一个形状为(?, 500)的张量。在numpy中，我会这样做： X[np.arange(len(X
浏览 23提问于2021-03-31得票数 2
回答已采纳
我刚刚注意到TensorFlow中一个意想不到的行为(至少对我来说是这样)。我以为tf.argmax (-argmin)从外到内都在张量的级别上运行，但显然它不是这样的？！示例：import tensorflow as tf
arr = np.array([[31,：input和dimension。因为数组arr的索引是arr[r
浏览 0提问于2016-06-29得票数 16
回答已采纳
我看不出我正在做的事情和正在运行的Google示例之间的区别，它的结构是我所遵循的。我做错了什么?我应该做的不一样吗？安装程序: Win 10 Home 64位20H2，Python3.7，TF2.4.1，TFP 0.12.2，在木星实验室运行
我一直在按照TFP 的示例逐步构建一个模型。用例1的代码运行良好，但我的并行模型没有，我看不到可能导致这种情况的区别。np.ndarray
浏览 0提问于2021-05-23得票数 1
回答已采纳
4回答
我知道tf.where将返回True值的位置，这样我就可以使用结果的shape[0]来获得True的数量。
然而，当我尝试使用它时，维度是未知的(这是有意义的，因为它需要在运行时计算)。所以我的问题是，我怎样才能访问一个维度，并在一个运算中像求和一样使用它？
浏览 3提问于2016-01-04得票数 46
回答已采纳
我有一个总共有9个元素的2D数组，生成一个3x3的表，如下所示：Z|Q|Z上表显示了一个无效的2D数组，因为第一列中有2个"Z“字符A|Q|Z上面显示了一个有效的二维数组，每个值在其各自的行<
浏览 13提问于2014-10-13得票数 0
1回答
我有一个PyTorch张量a，形状如下：a = torch.tensor([[[1., 0., 0., 0.]],
[[0., 1., 0., 0.]],张量a的每一行都有4个元素，1和0，假设I相应地索引这个张量的行和列。例如，行0(最上面的行)中的条目是[[1., 0., 0., 0.]]，而第3<
浏览 1提问于2020-08-09得票数 1
回答已采纳
我有个愚蠢的问题，但我不知道怎么解决它。我的输出层是一个完全连接的网络，它有一个二维加权矩阵W。假设W是一个3x3矩阵，我的算法需要W的行值是相同的。意思是在更新过程中，W更改为W'，但每一行的元素同时变化并保持相同
浏览 0提问于2018-07-29得票数 0
回答已采纳
我有一个电子表格，用来计算花在项目上的时间。我拿着发给我的每周时间表，挑选并记录特定项目中特定人员在任务上花费的小时数。我已经创建了一个数据透视表来处理大部分问题，但是我不能让数据透视表对行和列进行合计。(列总计很好)我可以手动对行进行求和，但是每次添加新任务时，我
浏览 3提问于2011-09-29得票数 0
扫码
添加站长 进交流群
领取专属 10元无门槛券
手把手带您无忧上云
相关资讯
热门标签
云服务器
ICP备案
实时音视频
对象存储
即时通信 IM活动推荐
腾讯云开发者
