整数的各数位变列表、排序字典、torch.where()、矩阵乘法

整数的各个数位上的数变成列表。

# Example integer
number = 12345

# Convert digits into an array using map() function
digits_array = list(map(int, str(number)))

print(digits_array) # [1, 2, 3, 4, 5]

排序字典

my_dict = {'key1': 30, 'key2': 10, 'key3': 50, 'key4': 20}

# Sort the dictionary by values in ascending order
sorted_dict = dict(sorted(my_dict.items(), key=lambda item: item[1]))

print(sorted_dict)

my_dict = {'key1': 30, 'key2': 10, 'key3': 50, 'key4': 20}

print(my_dict.items())
# dict_items([('key1', 30), ('key2', 10), ('key3', 50), ('key4', 20)])

torch.where()

import torch

# Create a tensor
tensor = torch.tensor([[1, 2, 3],
                       [4, 5, 6],
                       [7, 8, 9]])

# Apply torch.where() to find elements meeting a condition
result = torch.where(tensor > 5, tensor, torch.tensor(0))

print(result)
# tensor([[0, 0, 0],
        [0, 0, 6],
        [7, 8, 9]])

矩阵乘法

import torch

# Define two matrices
matrix1 = torch.tensor([[1, 2, 3],
                        [4, 5, 6]])

matrix2 = torch.tensor([[7, 8],
                        [9, 10],
                        [11, 12]])

# Perform matrix multiplication
result = torch.matmul(matrix1, matrix2)

print(result)
# tensor([[ 58,  64],
        [139, 154]])

import numpy as np

# Define two matrices as NumPy arrays
matrix1 = np.array([[1, 2],
                     [3, 4]])

matrix2 = np.array([[5, 6],
                     [7, 8]])

# Perform element-wise multiplication
result = matrix1 * matrix2

print(result)
# [[ 5 12]
 [21 32]]