向量元素间是否相等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]])
相关推荐
牵牛老人13 小时前
OpenCV学习探秘之二 :数字图像的矩阵原理,OpenCV图像类与常用函数接口说明,及其常见操作核心技术详解
opencv·学习·矩阵
shenghaide_jiahu14 小时前
数学建模——线性规划类题目(运筹优化类)
线性代数·数学建模
lqjun082716 小时前
相机内外参矩阵:从3D世界坐标到2D像素坐标变换
数码相机·3d·矩阵
恣艺17 小时前
LeetCode 1074:元素和为目标值的子矩阵数量
算法·leetcode·矩阵
Alfred king17 小时前
面试150 搜索二维矩阵
线性代数·矩阵·二分查找·数组
triticale1 天前
线性代数 下
线性代数
triticale1 天前
线性代数 上
线性代数
Alfred king2 天前
面试150 建立四叉树
矩阵··数组·分治
张欣-男2 天前
MIT线性代数02_矩阵消元
线性代数
石去皿2 天前
QKV 为什么是三个矩阵?注意力为何要除以 √d?多头注意力到底有啥用?
人工智能·线性代数·机器学习·矩阵