- 博客(3)
- 收藏
- 关注
RSS订阅原创 transfg
输入张量的尺寸为(batch_size, num_channels, height, width)，输出的最大值及其索引的尺寸为(batch_size, height-1, width-1)。last_map = last_map[:,:,0,1:] #对last_map进行切片操作，保留第0个维度和第1个维度，以及第3个维度从第1个元素开始的所有元素。#计算重叠补丁的数量。_, max_inx = last_map.max(2) #在last_map的第2个维度上找到最大值，并返回最大值及其索引。
2024-04-07 14:24:26
224
原创 vmamba
y = ys[:, 0] + ys[:, 1].view(B, -1, W, H).transpose(dim0=2, dim1=3).contiguous().view(B, -1, L) #：将上一步的结果重新排列并展平，然后返回。分割的维度取决于 channels_first 参数的值。ys = ys[:, 0:2] + ys[:, 2:4].flip(dims=[-1]).view(B, 2, -1, L) #将输出张量ys的前两个位置与翻转后的后两个位置相加，并重新调整形状。
2024-04-07 14:19:35
636
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人