LossMaskMatrix损失函数掩码矩阵

文章目录

  • [1. 理论](#1. 理论)
  • [2. python](#2. python)

1. 理论

A = [ 1 2 0 0 2 3 4 0 ] → B = [ 1 1 0 0 1 1 1 0 ] → C = [ 0.225 0.610 0 0 0.089 0.242 0.657 0 ] \begin{equation} A=\begin{bmatrix} 1&2&0&0\\\\ 2&3&4&0\end{bmatrix}\to B=\begin{bmatrix} 1&1&0&0\\\\ 1&1&1&0\end{bmatrix}\to C=\begin{bmatrix} 0.225&0.610&0&0\\\\ 0.089&0.242&0.657&0\end{bmatrix} \end{equation} A= 12230400 →B= 11110100 →C= 0.2250.0890.6100.24200.65700

2. python

python 复制代码
import torch
from torch import nn
from torch.nn import functional as F

torch.set_printoptions(precision=3, sci_mode=False)


class MatrixNoneZero2OnesLike(object):
    def __init__(self, in_matrix):
        self.in_matrix = in_matrix
        self._result = torch.zeros_like(self.in_matrix)

    @property
    def result(self):
        my_result = torch.zeros_like(self.in_matrix)
        my_result_bool = self.in_matrix.to(torch.bool)
        self._result = my_result.masked_fill(my_result_bool, 1)
        return self._result


class LossMaskedMatrix(object):
    def __init__(self, src_matrix):
        self.src_matrix = src_matrix.to(torch.float)
        self.nonzero = MatrixNoneZero2OnesLike(self.src_matrix)
        self.nonzero_matrix = self.nonzero.result
        self._loss_matrix = torch.zeros_like(self.src_matrix)

    @property
    def loss_matrix(self):
        my_soft_matrix = F.softmax(self.src_matrix, dim=-1)
        my_loss_matrix = my_soft_matrix * self.nonzero_matrix
        print(f"*" * 50)
        print(f"src_matrix=\n{self.src_matrix}")
        print(f"nonzero_matrix=\n{self.nonzero_matrix}")
        print(f"loss_matrix=\n{my_loss_matrix}")
        print(f"*" * 50)
        self._loss_matrix = my_loss_matrix

        return self._loss_matrix


if __name__ == "__main__":
    in_matrix = torch.tensor([[1, 2, 0, 0], [2, 3, 4, 0]]).to(torch.float)
    test_loss_matrix = LossMaskedMatrix(in_matrix)
    result = test_loss_matrix.loss_matrix
  • 结果:
python 复制代码
**************************************************
src_matrix=
tensor([[1., 2., 0., 0.],
        [2., 3., 4., 0.]])
nonzero_matrix=
tensor([[1., 1., 0., 0.],
        [1., 1., 1., 0.]])
loss_matrix=
tensor([[0.225, 0.610, 0.000, 0.000],
        [0.089, 0.242, 0.657, 0.000]])
**************************************************
相关推荐
SaleCoder39 分钟前
用Python构建机器学习模型预测股票趋势:从数据到部署的实战指南
开发语言·python·机器学习·python股票预测·lstm股票模型·机器学习股票趋势
AI街潜水的八角41 分钟前
图像修复:深度学习实现老照片划痕修复+老照片上色
人工智能·深度学习
机器学习之心7 小时前
三种深度学习模型(LSTM、CNN-LSTM、贝叶斯优化的CNN-LSTM/BO-CNN-LSTM)对北半球光伏数据进行时间序列预测
深度学习·cnn·lstm·cnn-lstm·贝叶斯优化的cnn-lstm
技术猿188702783517 小时前
实现“micro 关键字搜索全覆盖商品”并通过 API 接口提供实时数据(一个方法)
开发语言·网络·python·深度学习·测试工具
烛阴7 小时前
为什么你的Python项目总是混乱?层级包构建全解析
前端·python
三金C_C7 小时前
asyncio 与 uvloop
python·异步·asyncio
放飞自我的Coder7 小时前
【colab 使用uv创建一个新的python版本运行】
开发语言·python·uv
停走的风8 小时前
(李宏毅)deep learning(五)--learning rate
人工智能·深度学习·机器学习
黎茗Dawn8 小时前
连接new服务器注意事项
linux·python
LJianK18 小时前
Java和JavaScript的&&和||
java·javascript·python