pytorch除法之求元素在当前行所占比例

想念@思恋

于 2021-05-18 20:36:07 发布

阅读量405

点赞数

CC 4.0 BY-SA版权

分类专栏： pytorch python编程

本文链接：https://blog.youkuaiyun.com/tailonh/article/details/117000389

python编程同时被 2 个专栏收录

139 篇文章

订阅专栏

pytorch

47 篇文章

订阅专栏

本文通过一个具体示例介绍了如何使用Numpy和PyTorch进行矩阵操作及转换，包括创建矩阵、类型转换、归一化处理等过程。此教程适合初学者了解这两种工具的基本用法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import numpy as np
import torch
# max_seq_length = 3
# max_ngram_in_seq = 2
ngram_positions_matrix = np.array([[10, 12],
                                   [10, 3],
                                   [2, 3]])
ngram_positions_matrix = torch.from_numpy(ngram_positions_matrix.astype(np.float))
ngram_positions_matrix = torch.div(ngram_positions_matrix, torch.stack(
    [torch.sum(ngram_positions_matrix, 1)] * ngram_positions_matrix.size(1)).t() + 1e-10)
ngram_positions_matrix = ngram_positions_matrix.numpy()
print(ngram_positions_matrix)