向量元素间是否相等mask矩阵

文章目录

  • [1. 描述](#1. 描述)
  • [2. pytorch代码](#2. pytorch代码)

1. 描述

给定一个向量a_vector,生成一个mask矩阵用来判断向量元素间是否相等

假设有一个向量a_vector[1,2,1,4]此时就两个1相等,所以生成一个mask矩阵用来判断两个元素是否相等

2. pytorch代码

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

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

if __name__ == "__main__":
    run_code = 0
    a_vector = torch.tensor([1, 2, 1, 4]).reshape((-1, 1))
    print(f"a_vector=\n{a_vector}")
    a_matrix = a_vector - a_vector.T
    print(f"a_matrix=\n{a_matrix}")
    a_mask = a_matrix == 0
    print(f"a_mask=\n{a_mask}")
  • 结果:
python 复制代码
a_vector=
tensor([[1],
        [2],
        [1],
        [4]])
a_matrix=
tensor([[ 0, -1,  0, -3],
        [ 1,  0,  1, -2],
        [ 0, -1,  0, -3],
        [ 3,  2,  3,  0]])
a_mask=
tensor([[ True, False,  True, False],
        [False,  True, False, False],
        [ True, False,  True, False],
        [False, False, False,  True]])
相关推荐
峙峙峙5 小时前
线性代数--AI数学基础复习
人工智能·线性代数
我爱C编程6 小时前
基于拓扑结构检测的LDPC稀疏校验矩阵高阶环检测算法matlab仿真
算法·matlab·矩阵·ldpc·环检测
CVer儿8 小时前
svd分解求旋转平移矩阵
线性代数·算法·矩阵
张晓~1833994812113 小时前
数字人分身+矩阵系统聚合+碰一碰发视频: 源码搭建-支持OEM
线性代数·矩阵·音视频
山登绝顶我为峰 3(^v^)315 小时前
如何录制带备注的演示文稿(LaTex Beamer + Pympress)
c++·线性代数·算法·计算机·密码学·音视频·latex
微小冷1 天前
二关节机器人系统模型推导
线性代数·机器人·概率论·推导·拉格朗日函数·二关节机器人·机器人控制系统的设计
YuTaoShao2 天前
【LeetCode 热题 100】73. 矩阵置零——(解法二)空间复杂度 O(1)
java·算法·leetcode·矩阵
luofeiju2 天前
使用LU分解求解线性方程组
线性代数·算法
FF-Studio2 天前
【硬核数学 · LLM篇】3.1 Transformer之心:自注意力机制的线性代数解构《从零构建机器学习、深度学习到LLM的数学认知》
人工智能·pytorch·深度学习·线性代数·机器学习·数学建模·transformer
szekl2 天前
HDMI 2.0 4×2矩阵切换器412HN——多信号输入输出的高清解决方案
linux·矩阵·计算机外设·电脑·ekl