split function in pytorch medium code example

Example 1: map function in python

# Python program to demonstrate working 
# of map. 
  
# Return double of n 
def addition(n): 
    return n + n 
  
# We double all numbers using map() 
numbers = (1, 2, 3, 4) 
result = map(addition, numbers) 
print(list(result))

Example 2: function in python 3

#use the def keyword to declare a function in Python
def function(args,kwargs='attribute'):
    #here is the returned value
    return 'return value'