TensorFlow & PyTorch

1
2
3
4
5
6
7

# TensorFlow
import tensorflow as tf
model = tf.keras.Sequential([...])

# PyTorch
import torch
model = torch.nn.Sequential([...])

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20

# PyTorch tensor operations (conceptual)
# import torch

# Simulated tensor operations
def matmul(a, b):
    rows_a, cols_a = len(a), len(a[0])
    cols_b = len(b[0])
    result = [[0] * cols_b for _ in range(rows_a)]
    for i in range(rows_a):
        for j in range(cols_b):
            for k in range(cols_a):
                result[i][j] += a[i][k] * b[k][j]
    return result

A = [[1, 2], [3, 4]]
B = [[5, 6], [7, 8]]
C = matmul(A, B)
print("Matrix multiplication:")
for row in C:
    print(row)