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

最新推荐文章于 2024-09-19 21:18:19 发布

原创最新推荐文章于 2024-09-19 21:18:19 发布 · 419 阅读

0 ·

CC 4.0 BY-SA版权

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

143 篇文章

订阅专栏

pytorch

47 篇文章

订阅专栏

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

部署运行你感兴趣的模型镜像

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)