You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
35 lines
904 B
35 lines
904 B
import numpy as np
|
|
|
|
nd = np.array([[1, 2], [3, 4]])
|
|
|
|
# reshape
|
|
reveal_type(nd.reshape()) # E: ndarray
|
|
reveal_type(nd.reshape(4)) # E: ndarray
|
|
reveal_type(nd.reshape(2, 2)) # E: ndarray
|
|
reveal_type(nd.reshape((2, 2))) # E: ndarray
|
|
|
|
reveal_type(nd.reshape((2, 2), order="C")) # E: ndarray
|
|
reveal_type(nd.reshape(4, order="C")) # E: ndarray
|
|
|
|
# resize does not return a value
|
|
|
|
# transpose
|
|
reveal_type(nd.transpose()) # E: ndarray
|
|
reveal_type(nd.transpose(1, 0)) # E: ndarray
|
|
reveal_type(nd.transpose((1, 0))) # E: ndarray
|
|
|
|
# swapaxes
|
|
reveal_type(nd.swapaxes(0, 1)) # E: ndarray
|
|
|
|
# flatten
|
|
reveal_type(nd.flatten()) # E: ndarray
|
|
reveal_type(nd.flatten("C")) # E: ndarray
|
|
|
|
# ravel
|
|
reveal_type(nd.ravel()) # E: ndarray
|
|
reveal_type(nd.ravel("C")) # E: ndarray
|
|
|
|
# squeeze
|
|
reveal_type(nd.squeeze()) # E: ndarray
|
|
reveal_type(nd.squeeze(0)) # E: ndarray
|
|
reveal_type(nd.squeeze((0, 2))) # E: ndarray
|
|
|